/ML-Optimizers-JAX

Toy implementations of some popular ML optimizers using Python/JAX

Primary LanguagePython

Watchers