Scalable Distributional Robustness in a Class of Non-Convex Optimization with Guarantees