diffrax
by patrick-kidger (personal account)
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/ (description from the project’s GitHub page)
Repository facts
- Stars
- 2,114
- Forks
- 196
- Open issues and pull requests
- 260
- Primary language
- Python
- License
- Apache License 2.0 (Apache-2.0)
- Owner
- patrick-kidger · Personal account
- Created
- Jul 25, 2021
- Last push
- Sep 6, 2026
Topics
Add this badge to your README
Shows this repository’s rank in AI on GitHub and links back to this page. Ranks are by stars and change when the list is refreshed. The badge is not an endorsement or a rating. About the badges
Markdown (for a README)
[](https://unitedstatesofamericasuperintelligence.com/github/patrick-kidger--diffrax/)
HTML
<a href="https://unitedstatesofamericasuperintelligence.com/github/patrick-kidger--diffrax/"><img src="https://unitedstatesofamericasuperintelligence.com/badges/github/patrick-kidger--diffrax.svg" alt="Ranked #3,499 in AI on GitHub" height="20"></a>
Related projects
Projects that share this one’s GitHub topics.
Rank 2,628: patrick-kidger/equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
2,990 starsPythonApache-2.0Last push Sep 6, 2026
Rank 2,483: SciML/DifferentialEquations.jl
Multi-language suite for high-performance solvers of differential equations and scientific machine learning (SciML) components. Ordinary differential equations (ODEs), stochastic…
3,167 starsJuliaOther licenseLast push Oct 17, 2026
Rank 5,286: SciML/NeuralPDE.jl
Physics-Informed Neural Networks (PINN) Solvers of (Partial) Differential Equations for Scientific Machine Learning (SciML) accelerated simulation
1,235 starsJuliaOther licenseLast push Oct 17, 2026
Rank 6,399: SciML/DiffEqFlux.jl
Pre-built implicit layer architectures with O(1) backprop, GPUs, and stiff+non-stiff DE solvers, demonstrating scientific machine learning (SciML) and physics-informed machine lea…
929 starsJuliaMITLast push Oct 18, 2026
Rank 8,752: benmoseley/FBPINNs
Solve forward and inverse problems related to partial differential equations using finite basis physics-informed neural networks (FBPINNs)
577 starsPythonMITLast push May 29, 2026
Rank 4,421: DiffEqML/torchdyn
A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods
1,581 starsJupyter NotebookApache-2.0Last push May 2, 2024