DTensor, Correctness and the Costs of Abstraction
DTensor, PyTorch's distributed tensor abstraction, attaches placement metadata to every tensor to automatically propagate layouts and insert correct collective operations during distributed training. …