[P] Equinox (1.3k stars), a JAX library for neural networks and sciML

Hey folks! I wanted to advertise [Equinox](https://github.com/patrick-kidger/equinox) -- my now-surprisingly-popular ( :D ) JAX library for numerical models. These days that often means "neural networks", but I like to emphasise that this also includes ODEs/SDEs/linear solves, etc.

Here's the GitHub link: https://github.com/patrick-kidger/equinox

For those already using JAX, then Equinox is interesting because (a) it ships with a NN library, and (b) this is built around the idea that "everything is a pytree", which makes things easy to reason about and …

advanced easy errors everything jax library machinelearning pytree reason ships surgery tools true