bhadreshpsavani/LearningJax
Learning Notes and resources on Jax
This project is a personal learning journal and resource collection for the JAX framework, specifically for developers working with deep learning. It details the journey of understanding JAX's efficient NumPy operations, automatic differentiation, and parallel computation capabilities. Developers can follow along to see practical examples and comparisons with other frameworks like PyTorch and NumPy.
No commits in the last 6 months.
Use this if you are a developer looking for structured learning notes and code examples to understand and implement JAX for deep learning tasks.
Not ideal if you are a non-technical user seeking a ready-to-use application or a high-level overview without code examples.
Stars
7
Forks
—
Language
Jupyter Notebook
License
—
Category
Last pushed
Dec 08, 2021
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/bhadreshpsavani/LearningJax"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
explosion/thinc
🔮 A refreshing functional take on deep learning, compatible with your favorite libraries
google-deepmind/optax
Optax is a gradient processing and optimization library for JAX.
patrick-kidger/diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable....
google/grain
Library for reading and processing ML training data.
patrick-kidger/equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/