JAX: Accelerated machine-learning research via composable function transformations in Python | NeurIPS 2019 | Skye…

slideslive.comsite

JAX intro presentation in Program Transformations for Machine Learning workshop.

from

JAX

added

2026-10-10

likes

0

similar

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

   
2. ▲
   JAX on Cloud TPUs | NeurIPS 2020 | Skye Wanderman-Milne and James Bradbury drive.google.com

   site category #jax link ↗

   Presentation of TPU host access with demo.

3. ▲
   JAX github.com

   tool category #python-data-science link ↗

   Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more.

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

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. ▲
   Equinox github.com

   tool category #jax link ↗

   Callable PyTrees and filtered JIT/grad transformations => neural networks in JAX.

   

JAX › Videos: “JAX intro presentation in Program Transformations for Machine Learning workshop.”