Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/ - View it on GitHub