LoreonLabsPlatform

Overview

Intelligence

Markets
Builders
Research
Ecosystems
Launchpads

Search

Python

diffrax

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

PythonBuildingdeep-learningdifferential-equationsdynamical-systemsequinox

Stars

2.1K

Forks

183

Contributors

8

Last push

15d ago

Recent commits

Latest commits.

stop clipping rejected steps to t1 (#758)
2cd9890Jonathan Brodrick15d ago
fix reporting nonconvergence as divergence for non-Cauchy termination (#754)
9094d03Jonathan Brodrick1mo ago
Fix y_error returned from implicit Euler step when non-linear solve fails (#753)
31b2f16Haoyang Zhou1mo ago
make diffrax compatible with new lineax pre-release
7f0001cJonathan Brodrick2mo ago
standard dtype promotion
0f77ba0sammccallum2mo ago

fix backwards-in-time solve for srk's

12efb5bsammccallum2mo ago

Use dependency groups, uv, and local pre-commit hooks

56d78e1Patrick Kidger Bot2mo ago

bump pymdown-extensions; pin pygments

7ac1ea0Pixie Agent2mo ago

Top contributors

Builders behind this project.

simon-bachhuber