Class Central is learner-supported. When you buy through links on our site, we may earn an affiliate commission.
My get started with JAX repo
Stateful to stateless conversion
PyTrees in depth
Training an MLP in pure JAX
Custom PyTrees
Parallelism in JAX TPUs example
Communication between devices
value_and_grad and has_aux
Training an ML model on multiple machines
stop grad, per example grads
Implementing MAML in 3 lines
Outro
Aleksa Gordić - The AI Epiphany
Get personalized course recommendations, track subjects and courses with reminders, and more.
