patrick-kidger / equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
2,123Updated 2 weeks ago

Related projects

Alternatives and complementary repositories for equinox