JAX Crash Course - Accelerating Machine Learning Code

JAX Crash Course - Accelerating Machine Learning Code

AssemblyAI via YouTube Direct link

Intro & Outline

1 of 11

1 of 11

Intro & Outline

Class Central Classrooms beta

YouTube videos curated by Class Central.

Classroom Contents

JAX Crash Course - Accelerating Machine Learning Code

Automatically move to the next video in the Classroom when playback concludes

  1. 1 Intro & Outline
  2. 2 What is JAX
  3. 3 Speed comparison
  4. 4 Drop-in Replacement for NumPy
  5. 5 jit: just-in-time compiler
  6. 6 Limitations of JIT
  7. 7 grad: Automatic Gradients
  8. 8 vmap: Automatic Vectorization
  9. 9 pmap: Automatic Parallelization
  10. 10 Example Training Loop
  11. 11 What’s the catch?

Never Stop Learning.

Get personalized course recommendations, track subjects and courses with reminders, and more.

Someone learning on their laptop while sitting on the floor.