extra credit Search 85,062 technologies ⌘K Download for Mac

diffrax

GPU+autodiff-capable ODE/SDE/CDE solvers written in JAX.

Category
Python packages
pip install diffrax
↑↓ move ↵ open