{"library":"diffrax","type":"library","category":null,"description":"Diffrax is a high-performance Python library for solving ordinary, stochastic, and controlled differential equations (ODEs, SDEs, CDEs). Built on JAX, it offers GPU acceleration, automatic differentiation, and is designed for research in scientific machine learning. It is currently at version 0.7.2 and receives regular updates, often in sync with JAX ecosystem developments.","language":"python","status":"active","version":"0.7.2","tags":["JAX","ODE","SDE","CDE","numerical-solvers","differential-equations","autodiff","GPU","scientific-machine-learning"],"last_verified":"Tue May 26","install":[{"cmd":"pip install diffrax","imports":["from diffrax import diffeqsolve","from diffrax import ODETerm","from diffrax import Tsit5","from diffrax import Euler","from diffrax import SaveAt","from diffrax import Solution"]},{"cmd":"pip install 'diffrax[cuda]' # for GPU users","imports":[]}],"homepage":null,"github":"https://github.com/patrick-kidger/diffrax","docs":null,"changelog":null,"pypi":"https://pypi.org/project/diffrax/","npm":null,"openapi_spec":null,"status_page":null,"smithery":null,"compatibility":{"summary":{"python_range":"3.10–3.9","success_rate":50,"avg_install_s":12.6,"avg_import_s":4.27,"wheel_type":"wheel"},"url":"https://checklist.day/v1/registry/diffrax/compatibility"}}