3 ms·looks like flash attention concepts applied to kmeans, nice speedup resultsby matrix2596 7mo agolooks like flash attention concepts applied to kmeans, nice speedup results