torch-kmeans (pypi): PyTorch implementations of KMeans, Soft-KMeans and Constrained-KMeans which can be run on GPU and work on (mini-)batches of data.
Keywords: clustering|kmeans-clustering