Gallery of examples
Using torchtnt to count FLOPS
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
A simple Darcy-Flow spectrum analysis
A simple Darcy-Flow spectrum analysis
Training an FNO on Darcy-Flow
Training an FNO with incremental meta-learning
Training an FNO with incremental meta-learning