fix: Reduce K-means memory from O(n·k·d) to O(n·k) and fix empty cluster NaN#333

Merged

josevalim merged 3 commits intoelixir-nx:mainfrom

georgeguimaraes:fix/kmeans-memory-optimization

Apr 27, 2026

Commits on Apr 14, 2026

fix: Reduce K-means memory from O(n*k*d) to O(n*k) and fix empty cluster NaN
georgeguimaraes
committed

Commits on Apr 27, 2026

test: update k-means expected cluster ordering after empty-cluster fix
georgeguimaraes
committed
fix: restore inf for NaN distances in k-means++ initialization
georgeguimaraes
committed