Adds DiffusionUNet3D and its modules to physicsnemo.experimental#1616

Merged

CharlelieLrt merged 10 commits into

NVIDIA:mainfrom

CharlelieLrt:Implement-3d-diffusion-UNet

May 11, 2026

Commits on May 5, 2026

Adds DiffusionUNet3D and its modules to physicsnemo.experimental
CharlelieLrt
committed

Commits on May 10, 2026

Scope GPU tolerances to the 3D tests that need them
CharlelieLrt
committed
Add domain-parallel sanity tests for DiffusionUNet3D
CharlelieLrt
committed
Merge remote-tracking branch 'upstream/main' into Implement-3d-diffusion-UNet
CharlelieLrt
committed

Commits on May 11, 2026

Size domain-parallel test inputs for 8-GPU sharding
CharlelieLrt
committed