Black Friday 2024: 25+ Ways to Save on Online Learning
JAX Crash Course - Accelerating Machine Learning Code

JAX Crash Course - Accelerating Machine Learning Code

AssemblyAI via YouTube Direct link

jit: just-in-time compiler

5 of 11
Previous
Next

5 of 11

jit: just-in-time compiler

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.

Sign up for free
Someone learning on their laptop while sitting on the floor.