Non-Uniform Analysis For Non-Convex Optimization In Machine Learning