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

Visualizing computational fluid dynamics on a car

Visualizing computational fluid dynamics on a car

Data Generation

Examples demonstrating the use of our built-in finite difference solver to generate PDE data.

A simple finite-difference solver for the diffusion-advection equation

A simple finite-difference solver for the diffusion-advection equation

A finite difference solver for Burgers’ equation in 2 dimensions

A finite difference solver for Burgers' equation in 2 dimensions

Layers

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

Neighbor Search for Graph Neural Operators

Neighbor Search for Graph Neural Operators

Grid Embeddings

Grid Embeddings

Resampling layers

Resampling layers

Divergence-Free Spectral Projection

Divergence-Free Spectral Projection

DISCO Convolutions

DISCO Convolutions

Normalization Layers

Normalization Layers

Fourier Differentiation

Fourier Differentiation

Sinusoidal Embeddings

Sinusoidal Embeddings

Fourier Continuation

Fourier Continuation

Finite Differences

Finite Differences

Models

Examples showcasing our end-to-end NO architectures.

U-NO on Darcy-Flow

U-NO on Darcy-Flow

Training a SFNO on the spherical Shallow Water equations

Training a SFNO on the spherical Shallow Water equations

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