https://github.com/slimgroup/dfno

Distributed Fourier Neural Operators
https://github.com/slimgroup/dfno

fno mpi mpi4py python pytorch

Last synced: 14 days ago
JSON representation

Distributed Fourier Neural Operators

• Host: GitHub
• URL: https://github.com/slimgroup/dfno
• Owner: slimgroup
• License: mit
• Created: 2021-09-03T00:25:08.000Z (about 3 years ago)
• Default Branch: main
• Last Pushed: 2022-09-20T14:51:04.000Z (about 2 years ago)
• Last Synced: 2024-08-01T16:54:21.904Z (3 months ago)
• Topics: fno, mpi, mpi4py, python, pytorch
• Language: Python
• Size: 116 KB
• Stars: 28
• Watchers: 5
• Forks: 3
• Open Issues: 1
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
[![DOI](https://zenodo.org/badge/DOI/10.5281/zenodo.6981516.svg)](https://doi.org/10.5281/zenodo.6981516)

# Distributed Fourier Neural Operators

Implements a distributed version of [Fourier Neural Operators](https://arxiv.org/pdf/2010.08895.pdf)

based off of the [code](https://github.com/zongyi-li/fourier_neural_operator) written by Zongyi Li, Shuhao Cao, and Jack Griffiths

using [DistDL](https://github.com/distdl/distdl).

## Installation

To install this package you can either clone it and install it in dvelopper mode:

```bash

git clone https://github.com/slimgroup/dfno.git

cd dfno

pip install -e .

```

or directly install it with pip as a standard pythn package `pip install git+https://github.com/slimgroup/dfno.git`

## Author

This package is written by Thomas Grady  at Georgia Institute of Technology

## Citation

If you like and use our package, please cite our preprint:

```

@techreport {grady2022SCtll,

	title = {Model-Parallel Fourier Neural Operators as Learned Surrogates for Large-Scale Parametric PDEs},

	number = {TR-CSE-2022-1},

	year = {2022},

	month = {04},

	keywords = {CCS, deep learning, Fourier neural operators, HPC, large-scale, Model Parallelism, Multiphase Flow, Operator Learning},

	url = {https://arxiv.org/pdf/2204.01205.pdf},

	software = {https://github.com/slimgroup/dfno},

	author = {Thomas J. Grady II and Rishi Khan and Mathias Louboutin and Ziyi Yin and Philipp A. Witte and Ranveer Chandra and Russell J. Hewett and Felix J. Herrmann}

}

```