shreyansh26/ML-Optimizers-JAX

Toy implementations of some popular ML optimizers using Python/JAX

/ 100

Experimental

No commits in the last 6 months.

No License Stale 6m No Package No Dependents

Maintenance 0 / 25

Adoption 8 / 25

Maturity 8 / 25

Community 5 / 25

Stars

Forks

Language

Python

License

—

Category

Last pushed

Jun 20, 2021

Commits (30d)

Get this data via API

curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/shreyansh26/ML-Optimizers-JAX"

Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.

Higher-rated alternatives

metaopt/torchopt

TorchOpt is an efficient library for differentiable optimization built upon PyTorch.

SimplexLab/TorchJD

Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with...

clovaai/AdamP

AdamP: Slowing Down the Slowdown for Momentum Optimizers on Scale-invariant Weights (ICLR 2021)

nschaetti/EchoTorch

A Python toolkit for Reservoir Computing and Echo State Network experimentation based on...

gpauloski/kfac-pytorch

Distributed K-FAC preconditioner for PyTorch