Yogi Optimizer -

Try it on your next unstable training run. You might be surprised. 🚀

Yogi adds a tiny bit of compute per step and may need slightly more memory. In practice, it's negligible for most models. yogi optimizer

Beyond Adam: Meet Yogi – The Optimizer That Tames Noisy Gradients Try it on your next unstable training run

Developed by researchers at Google and Stanford, Yogi modifies Adam's adaptive learning rate mechanism to make it more robust to noisy gradients. yogi optimizer