Differential Equations Python Packages

torchsde

Differentiable SDE solvers with GPU support and efficient sensitivity analysis.

1.8M 2K 227

diffrax

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

252K 2K 186

torchcde

Differentiable controlled differential equation solvers for PyTorch with GPU support and memory-efficient adjoint backpropagation.

51K 484 50

brian2

Brian is a free, open source simulator for spiking neural networks.

41K 1K 287

cognitive-discovery-platform

Cognitive Discovery System — pure Python scientific computing. 18 modules: quantum, ML, NLP, graph theory, Monte Carlo, ODE solvers, FFT & signal filters, optimization, probability, statistics & time-series, symbolic math, numerical integration (1-D/2-D), knowledge graph. 1438 tests, 100% coverage, v1.2.3.

12K 12 1

jaxfne

JAX Tensor-Field Neural Equations (TFNE)

7K 2 0

semiflow-pde

Fast no_std Rust library for evolution equations and PDEs via Chernoff approximation of operator semigroups — heat, Schrödinger, diffusion, graph & manifold operators. Bindings for C, Python, and WebAssembly.

5K 2 0

heyoka

Python library for ODE integration via Taylor's method and LLVM

5K 89 14

cognitive-discovery-system

4K 12 1

diffeqpy

Solving differential equations in Python using DifferentialEquations.jl and the SciML Scientific Machine Learning organization

3K 606 52

zvode

Python bindings to the classic ZVODE library (in Fortran)

3K 3 0

pina-mathlab

Physics-Informed Neural networks for Advanced modeling

3K 771 107

pydiffgame

PyDiffGame is a Python implementation of a Nash Equilibrium solution to Differential Games, based on a reduction of Game Hamilton-Bellman-Jacobi (GHJB) equations to Game Algebraic and Differential Riccati equations, associated with Multi-Objective Dynamical Control Systems

2K 58 13

simframe

Python framework for setting up and running scientific simulations. It comes with a variety of integration schemes and some infrastructure of reading and writing data files.

2K 13 4

tedeous

TEDEouS - Torch Exhaustive Differential Equations Solver. Differential equation solver, based on pytorch library

1K 63 11

torchlaplace

Neural Laplace: Differentiable Laplace Reconstructions for modelling any time observation with O(1) complexity.

1K 84 12

diffraxtra

diffrax extras: OOP and vectorization

903 6 3

brian2cuda

A brian2 extension to simulate spiking neural networks on GPUs

899 77 22

neurodiffeq

A library for solving differential equations using neural networks based on PyTorch, used by multiple research groups around the world, including at Harvard IACS.

782 789 104