piwheels

diffrax

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

Installation

In a virtualenv (see these instructions if you need to create one):

pip3 install diffrax

Dependencies

• jaxtyping
• equinox
• typing-extensions
• jax
• optimistix
• lineax
• wadler-lindig

Releases

Version	Released	Bullseye Python 3.9	Bookworm Python 3.11	Trixie Python 3.13	Files
0.7.2	2026-02-18
diffrax-0.7.2-py3-none-any.whl (195 KB)
0.7.1	2026-02-01
diffrax-0.7.1-py3-none-any.whl (195 KB)
0.7.0	2025-03-14
diffrax-0.7.0-py3-none-any.whl (189 KB)
0.6.2	2024-12-24
diffrax-0.6.2-py3-none-any.whl (183 KB)
0.6.1	2024-12-10
diffrax-0.6.1-py3-none-any.whl (182 KB)
0.6.0	2024-07-01
diffrax-0.6.0-py3-none-any.whl (167 KB)
0.5.1	2024-05-19
diffrax-0.5.1-py3-none-any.whl (160 KB)
0.5.0	2024-01-08
diffrax-0.5.0-py3-none-any.whl (138 KB)
0.4.1	2023-08-03
diffrax-0.4.1-py3-none-any.whl (158 KB)
0.4.0	2023-05-22
diffrax-0.4.0-py3-none-any.whl (156 KB)
0.3.1	2023-02-23
diffrax-0.3.1-py3-none-any.whl (137 KB)
0.3.0	2023-02-21
diffrax-0.3.0-py3-none-any.whl (137 KB)
0.2.2	2022-11-15
diffrax-0.2.2-py3-none-any.whl (135 KB)
0.2.1	2022-08-03
diffrax-0.2.1-py3-none-any.whl (137 KB)
0.2.0	2022-07-20
diffrax-0.2.0-py3-none-any.whl (137 KB)
0.1.2	2022-05-18
diffrax-0.1.2-py3-none-any.whl (106 KB)
0.1.1	2022-04-07
diffrax-0.1.1-py3-none-any.whl (106 KB)
0.1.0	2022-03-30
diffrax-0.1.0-py3-none-any.whl (106 KB)
0.0.6	2022-03-29
diffrax-0.0.6-py3-none-any.whl (105 KB)
0.0.5	2022-03-21
diffrax-0.0.5-py3-none-any.whl (105 KB)
0.0.4	2022-02-20
diffrax-0.0.4-py3-none-any.whl (105 KB)
0.0.3	2022-02-15
diffrax-0.0.3-py3-none-any.whl (103 KB)
0.0.2	2022-02-10
diffrax-0.0.2-py3-none-any.whl (103 KB)
0.0.1	2022-02-07
diffrax-0.0.1-py3-none-any.whl (103 KB)

Issues with this package?

• Search issues for this package
• Package or version missing? Open a new issue
• Something else? Open a new issue