Introduction to JAX by Kevin Murphy

colab.research.google.comsite

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

from

JAX

added

2026-10-10

likes

0

similar

1. ▲
   Jax in Action manning.com

   site category #jax link ↗

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

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. ▲
   Using JAX to accelerate our research by David Budden and Matteo Hessel deepmind.com

   site category #jax link ↗

   Describes the state of JAX and the JAX ecosystem at DeepMind.

   
4. ▲
   Meta-Learning in 50 Lines of JAX by Eric Jang blog.evjang.com

   site category #jax link ↗

   Introduction to both JAX and Meta-Learning.

5. ▲
   Introduction to JAX youtube.com

   video category #jax link ↗

   Simple neural network from scratch in JAX.

   
6. ▲
   JAX: Accelerated Machine Learning Research | SciPy 2020 | VanderPlas youtube.com

   video category #jax link ↗

   JAX's core design, how it's powering new research, and how you can start using it.

   

JAX › Tutorials and Blog Posts: “Colab that introduces various aspects of the language and applies them to simple ML problems.”