Quick-Start

The neuralop library aims at providing you all the tools to easily use, build and train neural operators for your own applications and learn mapping between function spaces, in PyTorch.

Here are the main components of the library:

Module

Description

neuralop

Main library

neuralop.models

Full ready-to-use neural operators

neuralop.layers

Individual layers to build neural operators

neuralop.datasets

Convenience PyTorch data loaders for PDE datasets

neuralop.training

Utilities to train neural operators end-to-end