equinox

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

License

Apache License 2.0

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

Creator

patrick-kidger

Related apps

diffrax

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capab

Python1445apache-2.0

3 days ago

deep-learningdifferential-equationsdynamical-systems