SIGN IN SIGN UP

🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.

0 0 0 Python

Allow `device_mesh` have multiple dim (#38949)

* Feat: something

* Feat: initial changes

* tmp changes to unblock

* Refactor

* remove todo

* Feat: docstring

---------

Co-authored-by: Marc Sun <[email protected]>
M
Matej Sirovatka committed
82603b6cc284dbdf2b7a7cf070feb6a2c3bb53cf
Parent: 10c990f
Committed by GitHub <[email protected]> on 7/23/2025, 12:27:36 PM