/jaxde

Prototypes of differentiable differential equation solvers in JAX.

Primary LanguagePython

jaxde

Prototypes of differentiable differential equation solvers in JAX.