Getting started with JAX (MLPs, CNNs & RNNs) by Robert Lange

roberttlange.github.iosite

Neural network building blocks from scratch with the basic JAX operators.

from

JAX

added

2026-10-10

likes

0

similar

1. ▲
   Get started with JAX by Aleksa Gordić github.com

   tool category #jax link ↗

   A series of notebooks and videos going from zero JAX knowledge to building neural networks in Haiku.

   
2. ▲
   Evolving Neural Networks in JAX by Robert Tjarko Lange roberttlange.github.io

   site category #jax link ↗

   Explores how JAX can power the next generation of scalable neuroevolution algorithms.

3. ▲
   Learn JAX: From Linear Regression to Neural Networks by Rito Ghosh kaggle.com

   site category #jax link ↗

   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.

   
4. ▲
   Introduction to JAX youtube.com

   video category #jax link ↗

   Simple neural network from scratch in JAX.

   
5. ▲
   Jax in Action manning.com

   site category #jax link ↗

   A hands-on guide to using JAX for deep learning and other mathematically-intensive applications.

6. ▲
   Introduction to JAX by Kevin Murphy colab.research.google.com

   site category #jax link ↗

   Colab that introduces various aspects of the language and applies them to simple ML problems.

   

JAX › Tutorials and Blog Posts: “Neural network building blocks from scratch with the basic JAX operators.”