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
If you use TensorDiffEq in your work, please cite it via:
@article{mcclenny2021tensordiffeq,
title={TensorDiffEq: Scalable Multi-GPU Forward and Inverse Solvers for Physics Informed Neural Networks},
author={McClenny, Levi D and Haile, Mulugeta A and Braga-Neto, Ulisses M},
journal={arXiv preprint arXiv:2103.16034},
year={2021}
}
Thanks to our additional contributors:
@marcelodallaqua, @ragusa, @emiliocoutinho
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
File details
Details for the file tensordiffeq-0.2.0.tar.gz
.
File metadata
- Download URL: tensordiffeq-0.2.0.tar.gz
- Upload date:
- Size: 24.2 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/3.6.0 importlib_metadata/4.8.2 pkginfo/1.7.1 requests/2.26.0 requests-toolbelt/0.9.1 tqdm/4.62.3 CPython/3.10.0
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | d5ef4af1eb5b69ab1494d24836a6c5bc7ceb3d15c652d5729c42418b8e94d4d5 |
|
MD5 | 050214547bb6444e907715f1841095ce |
|
BLAKE2b-256 | cc75536a5d8617e188ad57c803192a5b18817dc72ef609db70c929b7554976d4 |
File details
Details for the file tensordiffeq-0.2.0-py3-none-any.whl
.
File metadata
- Download URL: tensordiffeq-0.2.0-py3-none-any.whl
- Upload date:
- Size: 24.5 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/3.6.0 importlib_metadata/4.8.2 pkginfo/1.7.1 requests/2.26.0 requests-toolbelt/0.9.1 tqdm/4.62.3 CPython/3.10.0
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | c0da8c4ac7782b6b14f0d68e9f5f16b9f10934f3f60dc93cb55c6f7686f01b4c |
|
MD5 | b72d0d13387bdfe1edecfcf3aa025070 |
|
BLAKE2b-256 | 9fd68ee4e1a6451760806d6e7ddd2844180805ae34cbbdcee48b0e30a8a747ef |