Reports
Created by
Created On
Last edited
Writing a Training Loop in JAX and Flax
In this article, we explore an end-to-end training and evaluation pipeline in JAX, Flax, and Optax for image classification, using W&B to track experiments.
8
geekyrakshit2022-07-13
JAXPiracy (Directed by Saurav and Soumik)
Comprehensive set of tutorials in JAX: Basics to Advanced
0
2022-07-29