- formatting
- images
- links
- math
- code
- blockquotes
- external-services
•
•
•
•
•
•
-
nanoJAXGPT — A pedagogical introduction to JAX/Equinox
Since its introduction, JAX has seen a significant rise in popularity within the Machine Learning (ML) community. A simple web search would reveal the vast community support, a variety of derivative projects, and a multitude of Python libraries built around JAX. This leads to the inevitable question — What is JAX, and why should I care?