NVIDIA Modulus
A Framework for Developing Physics Machine Learning Neural Network Models
NVIDIA Modulus is a neural network framework that blends the power of physics in the form of governing partial differential equations (PDEs) with data to build high-fidelity, parameterized surrogate models with near-real-time latency. Whether you’re looking to get started with AI-driven physics problems or designing digital twin models for complex non-linear, multi-physics systems, NVIDIA Modulus can support your work.

Accelerating Extreme Weather Prediction with FourCastNet
Siemens Energy HRSG Digital Twin Simulation Using NVIDIA Modulus and Omniverse
Maximizing Wind Energy Production Using Wake Optimization
Benefits
Scalable Performance
Solves larger problems faster by scaling from single-GPU to multi-node implementations.
AI Toolkit
Offers building blocks for developing physics machine learning surrogate models that combine both physics and data. The framework is generalizable to different domains and use cases—from engineering simulations to life sciences and from forward simulations to inverse/data assimilation problems.
Near-real-Time Inference
Provides parameterized system representation that solves for multiple scenarios in near real time, letting you train once offline to infer in real time repeatedly.
Easy to Adopt
Includes APIs for domain experts to work at a higher level of abstraction. Extensible to new applications with detailed reference applications serving as starting points.
Modulus Multi-GPU and Multi-Node Performance
NVIDIA Modulus supports multi-GPU and multi-node scaling using Horovod. This allows for multiple processes, each targeting a single GPU, with collective communication using the NVIDIA Collective Communications Library (NCCL) and Message Passing Interface (MPI).
This plot shows the weak scaling performance of Modulus on a field-programmable gate array (FPGA) test problem running on up to 32 NVIDIA V100 Tensor Core GPUs in four NVIDIA DGX™-1 systems. The scaling efficiency from one to 32 GPUs is more than 85 percent. This data was collected using Modulus v. 21.06.
Modulus Weak Scaling Across Multiple GPUs
Features
Modulus is a multi-physics framework that’s generalizable to multiple configurations enabled by parameterized geometry. With parameterized geometry, Modulus allows rapid design space exploration with single training for all configurations.
Training Pipeline for PhysicsML
Modulus provides a framework to model PDEs along with boundary conditions. It provides the end-to-end pipeline, from setting up the input tensor from geometry to training at scale.
Explicit Parameterization
It provides explicit parameter specifications for training the surrogate model with a range of values to learn for the design space and for inferring multiple scenarios simultaneously.
Novel Neural Network Architectures
It includes curated neural network architectures that are effective for physics-informed machine learning, such as Fourier feature networks, sinusoidal representation networks, or Fourier neural operators and adaptive Fourier neural operators.
For more details on Modulus (previously known as SimNet), please refer to NVIDIA SimNet: An AI-Accelerated Multi-Physics Simulation Framework. This paper reviews the neural network solver methodology, the Modulus architecture, and the features needed to effectively solve PDEs. It also includes real-world use cases that range from forward, parameterized, multi-physics simulations with turbulence and complex 3D geometries for industrial design optimization to inverse and data assimilation problems that aren’t addressed efficiently by traditional solvers.
Omniverse Integration
Available in v. 22.03
Modulus is now integrated with NVIDIA Omniverse™ via the Modulus extension that can be used to visualize the outputs of a Modulus-trained model. The extension enables you to import the output results into a visualization pipeline for common output scenarios, such as streamlines and iso-surfaces. It also provides an interface that enables interactive exploration of design variables and parameters for inferring new system behavior and visualizing it in near real time.
Omniverse integration documentation
What Others Are Saying
NVIDIA Modulus Resources

Using NVIDIA Modulus and Omniverse Wind Farm Digital Twin for Siemens Gamesa

Siemens Energy Taps NVIDIA to Develop Industrial Digital twin of Power Plant in Omniverse and Modulus
