JAX on Cloud TPUs | NeurIPS 2020 | Skye Wanderman-Milne and James Bradbury

drive.google.comsite

Presentation of TPU host access with demo.

from

JAX

added

2026-10-10

likes

0

similar

1. ▲
   JAX: Accelerated machine-learning research via composable function transformations in Python | NeurIPS 2019 | Skye… slideslive.com

   site category #jax link ↗

   JAX intro presentation in Program Transformations for Machine Learning workshop.

   
2. ▲
   SymJAX github.com

   tool category #jax link ↗

   Symbolic CPU/GPU/TPU programming.

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

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

   video · 24 min · Enthought category #jax link ↗

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

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

   video · 1 h 2 min · Google DeepMind category #jax link ↗

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

   
6. ▲
   Jumanji github.com

   tool category #jax link ↗

   A Suite of Industry-Driven Hardware-Accelerated RL Environments written in JAX.

JAX › Videos: “Presentation of TPU host access with demo.”