NVIDIA/physicsnemo v1.3.0

v1.3.0

Repository: NVIDIA/physicsnemo

Tag: v1.3.0

Published: 2025-11-18T06:50:15Z

Prerelease: no

Release notes: PhysicsNeMo General Release v1.3.0

Added

• Added mixture_of_experts for weather example in physicsnemo.examples.weather.

⚠️Warning: - It uses experimental DiT model subject to future API changes. Added some modifications to DiT architecture in physicsnemo.experimental.models.dit. Added learnable option to PositionalEmbedding in physicsnemo.models.diffusion.layers.

• Added lead-time aware training support to the StormCast example.
• Add a device aware kNN method to physicsnemo.utils.neighbors. Works with CPU or GPU

by dispatching to the proper optimized library, and torch.compile compatible.

• Added additional testing of the DoMINO datapipe.
• Examples: added a new example for full-waveform inversion using diffusion

models. Accessible in examples/geophysics/diffusion_fwi.

• Domain Parallelism: Domain Parallelism is now available for kNN, radius_search,

and torch.nn.functional.pad.

• Unified recipe for crash modeling, supporting Transolver and MeshGraphNet,

and three transient schemes.

• Added a check to stochastic_sampler that helps handle the EDMPrecond model,

which has a specific .forward() signature

• Added abstract interfaces for constructing active learning workflows, contained

under the physicsnemo.active_learning namespace. A preliminary example of how to compose and define an active learning workflow is provided in examples/active_learning. The moons example provides a minimal (pedagogical) composition that is meant to illustrate how to define the necessary parts of the workflow.

Changed

• Migrated Stokes MGN example to PyTorch Geometric.
• Migrated Lennard Jones example to PyTorch Geometric.
• Migrated physicsnemo.utils.sdf.signed_distance_field to a static return,

torch-only interface. It also now works on distributed meshes and input fields.

• Refactored DiTBlock to be more modular
• Added NATTEN 2D neighborhood attention backend for DiTBlock
• Migrated blood flow example to PyTorch Geometric.
• Refactored DoMINO model code and examples for performance optimizations and improved readability.
• Migrated HydroGraphNet example to PyTorch Geometric.
• Support for saving and loading nested physicsnemo.Modules. It is now

possible to create nested modules with m = Module(submodule, ...), and save and load them with Module.save and Module.from_checkpoint. ⚠️Warning: - The modules have to be physicsnemo.Modules, and not torch.nn.Modules.

• Support passing custom tokenizer, detokenizer, and attention Modules in

experimental DiT architecture

• Improved Transolver training recipe's configuration for checkpointing and normalization.

Fixed

• Set skip_scale to Python float in U-Net to ensure compilation works.
• Ensure stream dependencies are handled correctly in physicsnemo.utils.neighbors
• Fixed the issue with incorrect handling of files with consecutive runs of

combine_stl_solids.py in the X-MGN recipe.

• Fixed the RuntimeError: Worker data receiving interrupted error in the datacenter example.

Contributors

We’re grateful to everyone who contributed issues, feature ideas, fixes, and documentation updates — your input is what helps us continuously improve PhysicsNeMo for the whole community! A special shout-out to the authors of the pull requests listed above, in no particular order: @jleinonen , @Dibyajyoti-Chakraborty , @jialusui1102 , @abokov-nv , @melo-gonzo , @dran-dev , @RishikeshRanade , @swbg

Thank you :heart: — we truly appreciate your contributions and hope to see more from you in the future!