ESGD-M is a stochastic non-convex second order optimizer, suitable for training deep learning models, for PyTorch. - View it on GitHub