🫏 Mula

Machine Learning's Ultimate Learning Archive

A clean, educational implementation of Machine Learning's most influential works using JAX.

Models (WIP)

Projects

Project work is organized by year under projects/<year>/<project-name>/.

• projects/2025/examples/ contains the shared Colab and rendering examples.
• projects/2026/01_Linear_Regression_JAX/ contains the linear-regression project.
• projects/2026/shrink_clip/ contains the CLIP knowledge-distillation tutorial.

🏁 Getting Started (WIP)

---

🔗 Links

• JAX Documentation
• Flax
• Bonsai
• JAX AI Stack
• Learning-JAX