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