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

Communication between devices

7 of 12

7 of 12

Communication between devices

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.