v0.9.4

What's Changed

• docs: update to_batches batch_size documentation by @alexkohler in #1779
• chore: minor improvement over kmeans algorithm by @eddyxu in #1777
• perf: reduce torch memory allocation and avoid calculating y^2 multiple times in one iteration by @eddyxu in #1783
• chore: fix cuda-based indexing with f16 by @eddyxu in #1786
• fix: nan handling during training and dist accumulation by @eddyxu in #1787
• chore: revert async_dataset usage in cuda kmeans training by @eddyxu in #1788

New Contributors

• @alexkohler made their first contribution in #1779

Full Changelog: v0.9.3...v0.9.4