Arthur 20ce210ab7 Revert "remove dtensors, not explicit (#39840)" (#39912) ...

* Revert "remove dtensors, not explicit (#39840)"
This did not work with generation (lm_head needs extra care!)
This reverts commit 6dfd561d9c.

* update

* style?

2025-08-05 15:12:14 +02:00

..

test_tensor_parallel.py

Revert "remove dtensors, not explicit (#39840)" (#39912)

2025-08-05 15:12:14 +02:00