There's quite a few other libraries associated with Equinox in the JAX ecosystem: https://github.com/patrick-kidger/equinox?tab=readme-ov-file#see-also-other-l

There's quite a few other libraries associated with Equinox in the JAX ecosystem: https://github.com/patrick-kidger/equinox?tab=readme-ov-file#see-also-other-libraries-in-the-jax-ecosystem https://github.com/patrick-kidger/equinox?tab=readme-ov-file... I've enjoyed using Equinox and Diffrax for performing ODE simulations. To my knowledge the only other peer library with similar capabilities is the Julia DifferentialEquations.jl package.