Partial Differential Equations Python Packages

findiff

Python package for numerical derivatives and partial differential equations in any number of dimensions.

discretize

Discretization tools for finite volume and inverse problems.

fenics-ufl

UFL - Unified Form Language

py-pde

Python package for solving partial differential equations using finite differences.

simpeg

Simulation and Parameter Estimation in Geophysics - A python package for simulation and gradient based parameter estimation in the context of geophysical applications.

fipy

FiPy is a Finite Volume PDE solver written in Python

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.

fenics-ffcx

Next generation FEniCS Form Compiler for finite element forms

felupe

Open finite element infrastructure for nonlinear computational mechanics.

galerkin-transformer

[NeurIPS 2021] Galerkin Transformer: Neural Operator built on Attention for PDEs

fenics-ufl-legacy

UFL - Unified Form Language

pysdc

pySDC is a Python implementation of the spectral deferred correction (SDC) approach and its flavors, esp. the multilevel extension MLSDC and PFASST.

gelato

A Python library for Generalized Locally Toeplitz theory for Isogeometric Analysis.

torch-cfd

[ICLR 2025] Neural Operator-Assisted Computational Fluid Dynamics in PyTorch

pypde

A Python library for solving any system of hyperbolic or parabolic Partial Differential Equations. The PDEs can have stiff source terms and non-conservative components.

qc-grid

Python library for numerical integration, interpolation, and differentiation on (molecular) grids.

pyefvlib

A library used to solve physical problems governed by partial differential equations, in particular conservation equations using the Element-based Finite Volume Method.

rkstiff

Runge-Kutta adaptive-step solvers for nonlinear PDEs. Solvers include both exponential time differencing and integrating factor methods.

pdebench

PDEBench: An Extensive Benchmark for Scientific Machine Learning

triflow

Automatic finite difference discretization for 1D PDE with fast temporal solvers.

nops

nops: Neural Operators made simple — with physics-informed losses for FNO, DeepONet, and GNO

phlearn

The package phlearn for modelling pseudo-Hamiltonian systems by pseudo-Hamiltonian neural networks (PHNN), for ODEs and PDEs

poisson-cnn

Codebase for the Poisson CNN paper

hughes2d

A python package for simulating macroscopic pedestrian evacuations.