diffrax

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

• deep-learning
• differential-equations
• dynamical-systems
• equinox
• jax
• machine-learning
• neural-differential-equations
• neural-networks

Rank #3,499 of 9,507 in AI on GitHub, by stars. Data from the GitHub API, fetched Oct 18, 2026; star counts change daily. Stars measure popularity, not quality or safety, and a listing is not an endorsement. How this list is made

Related projects

Projects that share this one’s GitHub topics.

• 2,628

  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

• 2,483

  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

• 5,286

  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

• 6,399

  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

• 8,752

  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

• 4,421

  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