Jax¶
Tutorials¶
- A First Look at Jax - Madison May
- Getting Started with Jax (MLPs, CNNs & RNNs)- Robert Tjarko Lange
Flax * Flax: Google's First Open Source Approach to Flexibility in Machine Learning - Hackernoon
Haiku * Finetuning Transformers with Jax + Haiku - Madison May