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.

py-pde

Python package for solving partial differential equations using finite differences.

fenics-ufl

UFL - Unified Form Language

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

felupe

finite element analysis for continuum mechanics of solid bodies

fenics-ffcx

Next generation FEniCS Form Compiler for finite element forms

fenics-ufl-legacy

UFL - Unified Form Language

galerkin-transformer

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

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.

torch-cfd

PyTorch CFD

pyefvlib

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

pdebench

PDEBench: An Extensive Benchmark for Scientific Machine Learning

pysdc

A Python implementation of spectral deferred correction methods and the likes

nops

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

triflow

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

gelato

Generalized Locally Toeplitz

rkstiff

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

poisson-cnn

A convolutional neural network based Poisson solver

simpeg-archive

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

qc-grid

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

separable-operator-networks

Official repo for separable operator networks -- extreme-scale operator learning for parametric PDEs.

kpca-deeponet

Nonlinear model reduction for operator learning