PhysicsNeMo Models# PhysicsNeMo Models PhysicsNeMo Modules Basics Model Zoo How to Write Your Own PhysicsNeMo Model Converting PyTorch Models to PhysicsNeMo Models Saving and Loading PhysicsNeMo Models Constructing Nested Modules Backward Compatibility PhysicsNeMo Model Registry and Entry Points API Reference Fully Connected Network FullyConnected Fourier Neural Operators FNO FNO1DEncoder FNO2DEncoder FNO3DEncoder FNO4DEncoder AFNO AFNO2DLayer AFNOMlp Block PatchEmbed ModAFNO Graph Neural Networks MeshGraphNet MeshGraphNetProcessor Mesh_Reduced BiStrideMeshGraphNet Convolutional Networks Pix2Pix ResnetBlock Pix2PixUnet SRResNet ConvolutionalBlock3d PixelShuffle3d ResidualConvBlock3d SubPixel_ConvolutionalBlock3d Recurrent Neural Networks One2ManyRNN Seq2SeqRNN Operator Models DoMINO Domino Utils SMART (Experimental) AB-UPT (Experimental) Diffusion UNets SongUNet — The Primary Backbone Using UNet Backbones with the Diffusion Framework DhariwalUNet Lead-Time Aware Models Positional Embeddings (SongUNetPosEmbd) DiffusionUNet3D API Reference Diffusion Transformer (DiT) DiT Transolver Activation Checkpointing Building blocks FLARE and FLARE++ Activation Checkpointing GeoTransolver Activation Checkpointing Building blocks FLARE Attention Backend GLOBE GLOBE ISLA (Experimental) Citation Quick Start Weather / Climate Models DLWP HEALPixRecUNet GraphCastNet get_lat_lon_partition_separators() GraphCast Utils
