A-guide-to-ML-Workflows-with-JAX

A Talk on ML Workflows with JAX for TFUG Kolkata

The talk is split into two halves.

1. Evolution of JAX and its Power Tools by Aritra and Ritwik
2. A Simple Machine Learning Workflow in JAX and FLAX by Soumik

For reference follow the slide, presented by the authors.