Jax in Action

manning.comsite

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

from

JAX

added

2026-10-10

likes

0

similar

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

   
2. ▲
   GPJax github.com

   tool category #jax link ↗

   Gaussian processes in JAX.

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

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

5. ▲
   JAX Toolbox github.com

   tool category #jax link ↗

   Nightly CI and optimized examples for JAX on NVIDIA GPUs using libraries such as T5x, Paxml, and Transformer Engine.

   
6. ▲
   JAX, Flax & Transformers 🤗 github.com

   site category #jax link ↗

   3 days of talks around JAX / Flax, Transformers, large-scale language modeling and other great topics.

JAX › Books: “A hands-on guide to using JAX for deep learning and other mathematically-intensive applications.”