/FNO-in-JAX

Simple implementation of Fourier Neural Operators (FNOs) in the JAX deep learning framework together with Equinox.

Primary LanguageJupyter NotebookMIT LicenseMIT

No issues in this repository yet.