NVIDIA Modulus Core (Latest Release)

Modulus Tutorials

• Simple Training and Inference recipe
  
  • Basic Training workflow
  • Optimized Training workflow
  • Distributed Training workflow
  • Running inference on trained models
• Simple Logging and Checkpointing recipe
  
  • Logging in Modulus
  • Checkpointing in Modulus

Modulus API

• Modulus Models
  
  • Basics
  • Model Zoo
  • How to write your own Modulus model
  • Converting PyTorch Models to Modulus Models
  • Saving and Loading Modulus Models
  • Modulus Model Registry and Entry Points
• Modulus Datapipes
  
  • Benchmark datapipes
  • Weather and climate datapipes
  • Graph datapipes
  • CAE datapipes
• Modulus Metrics
  
  • Basics
  • General
  • Weather and climate metrics
• Modulus Deploy
  
  • ONNX
• Modulus Distributed
  
  • How does this work?
• Modulus Utils
  
  • Optimization utils
  • GraphCast utils
  • Filesystem utils
  • Generative utils
  • Geometry utils
  • Weather / Climate utils
• Modulus Launch Logging
  
  • Launch Logger
  • Console Logger
  • MLflow Logger
  • Weights and Biases Logger
  • Logging utils
• Modulus Launch Utils
  
  • Checkpointing

Examples: Weather and Climate

• ERA5 Data Downloader and Converter
• Adaptive Fourier Neural Operator (AFNO) for weather forecasting
• Deep Learning Weather Prediction (DLWP) model for weather forecasting
• GraphCast for weather forecasting
• Deep Learning Weather Prediction (DLWP-HEALPIX) model for weather forecasting
• Unified Recipe for Training Global Weather Forecasting Models

Examples: CFD

• AeroGraphNet for external aerodynamic evaluation
• MeshGraphNet for transient vortex shedding
• Temporal attention model in Mesh-Reduced space for transient vortex shedding
• Fourier Neural Operater for Darcy Flow
• Nested Fourier Neural Operater for Darcy Flow
• Physics Guided models for Darcy flow
• RNN for transient 2D Navier Stokes flow
• RNN for transient 3D Gray Scott system
• Physics Informed FNO for Nonlinear Shallow Water Equations

Examples: Healthcare

• MeshGraphNet for Reduced-Order cardiovascular simulations

Examples: Molecular Dynamics

• Molecular Dynamics using GNNs

Examples: Generative

• Generative Correction Diffusion Model (CorrDiff) for Km-scale Atmospheric Downscaling
• Problem overview
• Getting started
• Configs
• Dataset & Datapipe
• Training the models
• Logging
• References

Examples: Additive Manufacturing

• Virtual Foundary GraphNet

• Index