Skip to content

[JAX] GSPMD Deprecation Warning - Only trigger when the primitive is invoked#2729

Merged
phu0ngng merged 7 commits intoNVIDIA:mainfrom
phu0ngng:gspmd
Mar 4, 2026
Merged

[JAX] GSPMD Deprecation Warning - Only trigger when the primitive is invoked#2729
phu0ngng merged 7 commits intoNVIDIA:mainfrom
phu0ngng:gspmd

Commits

Commits on Mar 3, 2026

Commits on Mar 4, 2026