transformers
PATCH: add back n-dim device-mesh + fix tp trainer saving
#39693
Merged

Loading