https://mytecharm.com.co/post/batched-k-means-with-python-numba-and-cuda-c-iz2c88