Gallery of examples

A gallery of interactive examples, demonstrating use cases of the library.

Using torchtnt to count FLOPS

Using torchtnt to count FLOPS

A simple Darcy-Flow dataset

A simple Darcy-Flow dataset

Checkpointing and loading training states

Checkpointing and loading training states

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

A simple Darcy-Flow spectrum analysis

A simple Darcy-Flow spectrum analysis

Training an FNO on Darcy-Flow

Training an FNO on Darcy-Flow

Training an FNO with incremental meta-learning

Training an FNO with incremental meta-learning

Visualization of discrete-continuous convolutions

Visualization of discrete-continuous convolutions

Gallery generated by Sphinx-Gallery