JAX

A collection of YouTube about JAX.

JAX in 100 Seconds

@Fireship, June 2024
Neural Networks in Equinox (JAX DL framework) with Optax

@MachineLearningSimulation, June 2023
JAX Foundations

@GoogleResearch, January 2023
Introduction to JAX

@googlecloudtech, May 2020