Differential Equations Python Packages

torchsde

Differentiable SDE solvers with GPU support and efficient sensitivity analysis.

diffrax

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

torchcde

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

brian2

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

heyoka

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

simframe

Framework for Scientific Simulations

pina-mathlab

Physics-Informed Neural networks for Advanced modeling

tedeous

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

torchlaplace

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

diffeqpy

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

brian2cuda

A Brian2 extension to simulate spiking neural networks on GPUs

solve-dae

Python implementation of solvers for differential algebraic equation's (DAE's) that should be added to scipy one day.

scipy-dae

Python implementation of solvers for differential algebraic equation's (DAE's) that should be added to scipy one day.

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.

diffraxtra

diffrax extras: OOP and vectorization

arflab

Python library for creating a mathematical environment

tensordiffeq

Efficient and Scalable Physics-Informed Deep Learning and Scientific Machine Learning on top of Tensorflow for multi-worker distributed computing

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

dqbee

(TACAS 24) Python Package for disspaitve quadratization, from Paper "Dissipative quadratizations of polynomial ODE systems"

unimath

Python library for creating a mathematical environment

odeintw

odeintw provides a wrapper of scipy.integrate.odeint that allows it to handle complex and matrix differential equations.

tantheta

tantheta is a versatile Python library for symbolic computation and problem solving across mathematics, physics, and chemistry!

wavetorch

🌊 Numerically solving and backpropagating through the wave equation

paramfittorchdemo

How to use pytorch to fit parameters of differential equations