Learn JAX: From Linear Regression to Neural Networks by Rito Ghosh
A gentle introduction to JAX and using it to implement Linear and Logistic Regression, and Neural Network models and using them to solve real world problems.
- from
- JAX
- added
- 2026-10-10
- likes
- 0
similar
-
Getting started with JAX (MLPs, CNNs & RNNs) by Robert Lange roberttlange.github.io
Neural network building blocks from scratch with the basic JAX operators.
-
-
Get started with JAX by Aleksa Gordić github.com
A series of notebooks and videos going from zero JAX knowledge to building neural networks in Haiku.
-
Evolving Neural Networks in JAX by Robert Tjarko Lange roberttlange.github.io
Explores how JAX can power the next generation of scalable neuroevolution algorithms.
-
Jax in Action manning.com
A hands-on guide to using JAX for deep learning and other mathematically-intensive applications.
-
Meta-Learning in 50 Lines of JAX by Eric Jang blog.evjang.com
Introduction to both JAX and Meta-Learning.
JAX › Tutorials and Blog Posts: “A gentle introduction to JAX and using it to implement Linear and Logistic Regression, and Neural Network models and using them to solve real world problems.”