torch.distributed kmeans #1301

Job	Run time
Format	45s
Linux x86_64 GPU packages (CUDA 11.4.4)	0s
Linux x86_64 (cmake)	13m 35s
Linux arm64 packages	0s
Linux x86_64 GPU packages (CUDA 12.1.1)	0s
Linux x86_64 GPU w/ RAFT packages (CUDA 11.8.0)	0s
Linux x86_64 GPU w/ RAFT packages (CUDA 12.1.1)	0s
Linux x86_64 packages	0s
OSX arm64 packages	0s
Windows x86_64 packages	0s
Linux arm64 (conda)	11m 11s
Linux x86_64 (conda)	13m 20s
Linux x86_64 AVX2 (cmake)	13m 16s
Linux arm64 SVE (cmake)	40m 10s
Linux x86_64 AVX512 (cmake)	20m 29s
Linux x86_64 GPU (cmake)	35m 6s
Linux x86_64 GPU w/ RAFT (cmake)	1h 34m 59s
Linux x86_64 GPU w/ ROCm (cmake)	28m 42s
Windows x86_64 (conda)	38m 22s
	5h 9m 55s

Provide feedback