#Equinox
Showing 3 of 3 repositories tagged #equinox, ranked by stars
patrick-kidger
equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
Score
100
★ 2.9k
⑂ 204
+4/day
Python
patrick-kidger
diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
Score
50
★ 2.1k
⑂ 186
+3/day
Python
patrick-kidger
optimistix
Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
Score
0
★ 600
⑂ 54
+2/day
Python