ysngshn/ivon-optax

An Optax-based JAX implementation of the IVON optimizer for large-scale VI training of NNs (ICML'24 spotlight)

/ 100

Experimental

No commits in the last 6 months.

Stale 6m No Package No Dependents

Maintenance 0 / 25

Adoption 5 / 25

Maturity 9 / 25

Community 0 / 25

Stars

Forks

—

Language

Python

License

GPL-3.0

Category

Last pushed

Dec 19, 2024

Commits (30d)

Get this data via API

curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/ysngshn/ivon-optax"

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

Higher-rated alternatives

google-deepmind/optax

Optax is a gradient processing and optimization library for JAX.

patrick-kidger/equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

explosion/thinc

🔮 A refreshing functional take on deep learning, compatible with your favorite libraries

google/grain

Library for reading and processing ML training data.

extropic-ai/thrml

Thermodynamic Hypergraphical Model Library in JAX