DeepXDE

DeepXDE is a deep learning library for solving differential equations on top of TensorFlow.

Use DeepXDE if you need a deep learning library that

  • solves partial differential equations (PDEs),
  • solves integro-differential equations (IDEs),
  • solves fractional partial differential equations (fPDEs),
  • solves inverse problems for differential equations,
  • approximates functions from a dataset with/without constraints,
  • approximates functions from multi-fidelity data,
  • approximates operators.

DeepXDE is extensible to solve other problems in Scientific Machine Learning.

Papers

  • DeepXDE: arXiv
  • Multi-fidelity neural network: arXiv
  • Deep operator network (DeepONet): arXiv

Features

DeepXDE supports

  • complex domain geometries without tyranny mesh generation. The primitive geometries are interval, triangle, rectangle, polygon, disk, cuboid, and sphere. Other geometries can be constructed as constructive solid geometry (CSG) using three boolean operations: union, difference, and intersection;
  • multi-physics, i.e., coupled PDEs;
  • 5 types of boundary conditions (BCs): Dirichlet, Neumann, Robin, periodic, and a general BC;
  • time-dependent PDEs are solved as easily as time-independent ones by only adding initial conditions;
  • residual-based adaptive refinement (RAR);
  • uncertainty quantification using dropout;
  • two types of neural networks: fully connected neural network, and residual neural network;
  • many different losses, metrics, optimizers, learning rate schedules, initializations, regularizations, etc.;
  • useful techniques, such as dropout and batch normalization;
  • callbacks to monitor the internal states and statistics of the model during training;
  • enables the user code to be compact, resembling closely the mathematical formulation.

All the components of DeepXDE are loosely coupled, and thus DeepXDE is well-structured and highly configurable. It is easy to customize DeepXDE to meet new demands.

User guide

API reference

If you are looking for information on a specific function, class or method, this part of the documentation is for you.

Indices and tables