Neural Operator Python Packages

poraque

For electronic structure calculations with neural operators

autofd

Automatic Functional Differentiation in JAX

flashslice

Fused Triton kernels for Transolver's slice/deslice. Official code for "Does Transolver really need a Transformer?" (arXiv:2609.32525)

torch-cfd

[ICLR 2025] Neural Operator-Assisted Computational Fluid Dynamics in PyTorch

galerkin-transformer

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

nops

nops is designed to be an elegant, high-level layer over complex neural operator architectures

koopmanlab

A library for Koopman Neural Operator with Pytorch.