← Latest products

PyTorch K-Means GPU-friendly

single-file, hierarchical and resampling. I built a small, self-contained K-Means implementation in pure PyTorch: https://gitlab.com/hassonofer/pt_kmeans I was working on dataset sampling and approximate nearest neighbor search, and tried several existing libraries for large-scale K-Means.

Visit news.ycombinator.comexternal site

Reactions (1)

Similar products

by description similarity

  1. Open-Sourcing Google's Lattice Models in PyTorch

    github.com

    A PyTorch implementation of constrained optimization and modeling techniques

    Hacker News13 pts

  2. I implemented the Kimi K3 paper from scratch in PyTorch

    github.com

    Independent from-scratch implementation of the Kimi K3 architecture (arXiv:2607.24653v1): KDA, NoPE, latent-space MoE, and the systems co-designs. Table 1 reproduced to 0.09%.

    Hacker News10 pts

  3. Want something better than k-means? Try BanditPAM

    github.com

    BanditPAM C++ implementation and Python package

    Hacker News281 pts

  4. High speed graphics rendering research with tinygrad/tinyJIT

    github.com

    You like pytorch? You like micrograd? You love tinygrad! ❤️

    Hacker News31 pts

  5. Cant

    github.com

    Library written in Rust that provides PyTorch-like functionality

    Hacker News50 pts