Skip to content

ShardTensor: fixes for torch.compile with domain parallelism (#1986) - #2210

Draft
coreyjadams wants to merge 8 commits into
mainfrom
dp-06-compile-1986
Draft

ShardTensor: fixes for torch.compile with domain parallelism (#1986)#2210
coreyjadams wants to merge 8 commits into
mainfrom
dp-06-compile-1986

NATTEN: route subclass inputs through physicsnemo::na{1,2,3}d ops

93dc28f
Select commit
Loading
Failed to load commit list.

Select a check to view from the sidebar