warner-benjamin / optimi

Fast, Modern, Memory Efficient, and Low Precision PyTorch Optimizers
https://optimi.benjaminwarner.dev
MIT License
53 stars 2 forks source link