Skip to content

Jax kmeans implementation, sklearn removal from main deps - #461

Open
kylejcaron wants to merge 16 commits into
probml:mainfrom
kylejcaron:jax-kmeans
Open

Jax kmeans implementation, sklearn removal from main deps#461
kylejcaron wants to merge 16 commits into
probml:mainfrom
kylejcaron:jax-kmeans

Expose n_local_trials on kmeans; lower n_init default to 3

5d43625
Select commit
Loading
Failed to load commit list.
Sign in for the full log view

Annotations

1 warning
ubuntu-latest with Python 3.12
succeeded Aug 26, 2026 in 8m 44s