patrick-kidger / equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
2,098Updated last week

Related projects

Alternatives and complementary repositories for equinox