Distributed PDE Solver in Tensorflow
Project description
Efficient and Scalable Physics-Informed Deep Learning
Collocation-based PINN PDE solvers for prediction and discovery methods on top of Tensorflow 2.X for multi-worker distributed computing.
Use TensorDiffEq if you require:
- A meshless PINN solver that can distribute over multiple workers (GPUs) for forward problems (inference) and inverse problems (discovery)
- Scalable domains - Iterated solver construction allows for N-D spatio-temporal support
- support for N-D spatial domains with no time element is included
- Self-Adaptive Collocation methods for forward and inverse PINNs
- Intuitive user interface allowing for explicit definitions of variable domains, boundary conditions, initial conditions, and strong-form PDEs
What makes TensorDiffEq different?
-
Completely open-source
-
Self-Adaptive Solvers for forward and inverse problems, leading to increased accuracy of the solution and stability in training, resulting in less overall training time
-
Multi-GPU distributed training for large or fine-grain spatio-temporal domains
-
Built on top of Tensorflow 2.0 for increased support in new functionality exclusive to recent TF releases, such as XLA support, autograph for efficent graph-building, and grappler support for graph optimization* - with no chance of the source code being sunset in a further Tensorflow version release
-
Intuitive interface - defining domains, BCs, ICs, and strong-form PDEs in "plain english"
*In development
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for tensordiffeq-0.1.6.7-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 4d6ec7a31c2d8a963120c102efd6e068776e14d57f0c64052267b2aaa4255696 |
|
MD5 | be32a7bc6e92c4a3e30a23d7f85ec704 |
|
BLAKE2b-256 | 04ad7c4326a8d4e6a01c8c04e8a230000aaec35c7b79a0453355cec773d69d8d |