Machine Learning with JAX - From Hero to HeroPro+

Machine Learning with JAX - From Hero to HeroPro+

Aleksa Gordić - The AI Epiphany via YouTube Direct link

stop grad, per example grads

10 of 12

10 of 12

stop grad, per example grads

Class Central Classrooms beta

YouTube videos curated by Class Central.

Classroom Contents

Machine Learning with JAX - From Hero to HeroPro+

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

  1. 1 My get started with JAX repo
  2. 2 Stateful to stateless conversion
  3. 3 PyTrees in depth
  4. 4 Training an MLP in pure JAX
  5. 5 Custom PyTrees
  6. 6 Parallelism in JAX TPUs example
  7. 7 Communication between devices
  8. 8 value_and_grad and has_aux
  9. 9 Training an ML model on multiple machines
  10. 10 stop grad, per example grads
  11. 11 Implementing MAML in 3 lines
  12. 12 Outro

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.