astanziola / fourier-neural-operator-flax

An unofficial implementation of the Fourier Neural Operator in Flax

Fourier Neural Operator (Flax)

This is an unofficial Flax (JAX) port of the Fourier Neural Operator developed by Zongyi Li et al.

Please visit this repository to access the official PyTorch implementation of FNO form the paper's authors, and for the citation policy in case you use FNO in your research.

Requirements

To install fno, you just need to have flax installed (see the requirements.txt file).

Then:

• To generate the data you need MATLAB (or, probably, Octave: haven't tested it).
• To run the training scripts, you'll need to install the packages in requirements-train.txt, for example using pip install -r requirements-train.txt.