Skip to content
This repository has been archived by the owner. It is now read-only.

Combines power of torch, numerical methods to conquer and solve ALL {O,P}DEs

License

Notifications You must be signed in to change notification settings

aminevdom/torch_DE_solver

 
 

Repository files navigation

TEDEouS - Torch Exhaustive Differential Equation Solver

The purpose of the project

  1. Make equation discovery more transparent and illustrative
  2. Combine power of pytorch, numerical methods and math overall to conquer and solve ALL {O,P}DEs. There are some examples to provide a little insight to an operator form

Table of Contents

Core features

  • Solve ODE initial or boundary-value problems
  • Solve PDE initial-boundary value problems
  • Use variable models and their differentiation methods
  • Faster solution using cache

Installation

TEDEouS can be installed with pip:

$ git clone https://github.com/ITMO-NSS-team/torch_DE_solver.git
$ cd torch_DE_solver
$ pip install -r requirements.txt

Examples

After the TEDEouS is installed the user may refer to various examples that are in examples forlder.

$ cd examples

Every example is designed such that the boxplots of the launches are commented and the preliminary results are not shown, but stored in separate folders.

  • Legendre polynomial equation
$ python example_ODE_Legendre.py

or

$ python example_ODE_Legendre_autograd.py
  • Panleve transcendents (others are places in 'examples\to_renew' folder due to the architecture change)
$ python example_Painleve_I.py
  • Wave equation (non-physical conditions for equation discovery problem)
$ python example_wave_paper_autograd.py
  • Wave equation (initial-boundary value problem)
$ python example_wave_physics.py
  • Heat equation
$ python example_heat.py
  • KdV equation (non-physical conditions for equation discovery problem)
$ python example_KdV.py
  • KdV equation (solitary solution with periodic boundary conditions)
$ python example_KdV_periodic.py
  • Burgers equation and DeepXDE comparison
$ python WIP

Project Structure

Stable version is located in the master branch.

Documentation

https://torch-de-solver.readthedocs.io/en/docs/index.html

Getting started

Schroedinger equation example step-by-step https://torch-de-solver.readthedocs.io/en/docs/tedeous/examples/schrodinger.html

License

TEDEouS is distributed under BSD-3 licence found in LICENCE file

Contacts

  • Feel free to make issues or contact @SuperSashka directly

Citation

@article{"name",
title = {}, author = {}, journal = {}, year = {}, issn = {}, doi = {}}

About

Combines power of torch, numerical methods to conquer and solve ALL {O,P}DEs

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 100.0%