Code for Imitating Deep Learning Dynamics via Locally Elastic Stochastic Differential Equations - View it on GitHub