Introduction to JAX

youtube.comvideo

Simple neural network from scratch in JAX.

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. ▲
   Getting started with JAX (MLPs, CNNs & RNNs) by Robert Lange roberttlange.github.io

   site category #jax link ↗

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

3. ▲
   NeurIPS 2020: JAX Ecosystem Meetup youtube.com

   video category #jax link ↗

   JAX, its use at DeepMind, and discussion between engineers, scientists, and JAX core team.

   
4. ▲
   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.

5. ▲
   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.

   
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 › Videos: “Simple neural network from scratch in JAX.”