Neural Operator Python Packages

galerkin-transformer

[NeurIPS 2021] Galerkin Transformer: Neural Operator built on Attention for PDEs

torch-cfd

PyTorch CFD

nops

nops: Neural Operators made simple — with physics-informed losses for FNO, DeepONet, and GNO

autofd

Automatic Functional Differentiation in JAX

koopmanlab

A library for Koopman Neural Operator with Pytorch.