patrick-kidger / equinox

git clone https://github.com/patrick-kidger/equinox.git

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