simipixel / diffrax Goto Github PK
View Code? Open in Web Editor NEWThis project forked from patrick-kidger/diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
License: Apache License 2.0