Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
Why do you think that https://github.com/lululxvi/deepxde is a good alternative to diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
Why do you think that https://github.com/lululxvi/deepxde is a good alternative to diffrax