JAX and Equinox: What are they and why should I bother?

This post is written as a Jupyter notebook which you can run and edit using the link below: