matomatical/hijax
An introduction to vanilla JAX for deep learning research
This course helps deep learning researchers build, train, and understand neural networks more efficiently. It takes your existing Python and NumPy programming skills and teaches you how to use JAX to accelerate computations and write more elegant, functional code. The end result is faster experimentation and development of deep learning models.
Use this if you are a deep learning scientist or researcher familiar with Python and NumPy, who wants to leverage JAX for faster and more elegant neural network development and experimentation.
Not ideal if you are new to deep learning concepts, Python programming, or prefer a framework with a larger, more mature ecosystem like PyTorch.
Stars
13
Forks
4
Language
Python
License
MIT
Category
Last pushed
Mar 02, 2026
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/matomatical/hijax"
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/