The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series of tutorials (YouTube videos and Jupyter Notebooks) as well a…

This repos contains notebooks for the Advanced Solutions Lab: ML Immersion

An LLM playground you can run on your laptop