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

1 min read Original article ↗

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

Open In Colab