gordicaleksa/get-started-with-JAX
The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series of tutorials (YouTube videos and Jupyter Notebooks) as well as the content I found useful while learning about the JAX ecosystem.
776 stars. No commits in the last 6 months.
Stars
776
Forks
117
Language
Jupyter Notebook
License
MIT
Category
Last pushed
Nov 29, 2023
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/gordicaleksa/get-started-with-JAX"
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