K means / Spherical ( Angular) K means

Is there any implementation of K-means specifically Spherical K-means in pytorch? Thanks.