Tensors and Dynamic neural networks in Python with strong GPU acceleration - Allow parallelize_module to get device_mesh from ambient context · pytorch/pytorch@31715be
Tensors and Dynamic neural networks in Python with strong GPU acceleration - Update on "Allow parallelize_module to get device_mesh from ambient c… · pytorch/pytorch@47952c0
Tensors and Dynamic neural networks in Python with strong GPU acceleration - Allow parallelize_module to get device_mesh from ambient context · pytorch/pytorch@cda431d
Tensors and Dynamic neural networks in Python with strong GPU acceleration - Allow parallelize_module to get device_mesh from ambient context · pytorch/pytorch@9db407d
Tensors and Dynamic neural networks in Python with strong GPU acceleration - Allow parallelize_module to get device_mesh from ambient context · pytorch/pytorch@47952c0
Tensors and Dynamic neural networks in Python with strong GPU acceleration - Tensor Parallel (TP) broken on 2.6 (cannot `parallelize_module` correctly) · pytorch/pytorch@519f547