Examples

A gallery of interactive examples that showcase how the tools we provide in neuraloperator can be applied to a variety of problems. Check out the User Guide for more detailed information on the theory behind neural operators.

Data

Examples of NO layers in action.

A simple Darcy-Flow dataset

A simple Darcy-Flow dataset

A simple Darcy-Flow spectrum analysis

A simple Darcy-Flow spectrum analysis

Layers

Examples of individual layers which comprise operators or parts of operators for composition into end-to-end models.

Visualizing neighbor search

Visualizing neighbor search

Grid embeddings

Grid embeddings

Visualizing discrete-continuous convolutions

Visualizing discrete-continuous convolutions

Models

Examples showcasing our end-to-end NO architectures.

Training a SFNO on the spherical Shallow Water equations

Training a SFNO on the spherical Shallow Water equations

U-NO on Darcy-Flow

U-NO on Darcy-Flow

Training an FNO on Darcy-Flow

Training an FNO on Darcy-Flow

Training and Meta-Algorithms

Examples to highlight our flexible Trainer module and the meta-algorithms built on top of it.

Using torchtnt to count FLOPS

Using torchtnt to count FLOPS

Checkpointing and loading training states

Checkpointing and loading training states

Training an FNO with incremental meta-learning

Training an FNO with incremental meta-learning

Gallery generated by Sphinx-Gallery