Yujun Lin

Berkeley AI Research (BAIR)

Papers on Lattice

Total citations

Topics

h-index

Publication activitypapers/week, last 8 weeks

Research focus

Distributed Systems & Hardware (1)Inference & Quantization (1)Training Efficiency & Optimization (1)

Frequent co-authors

Shuo Yang (1)Haocheng Xi (1)Yilong Zhao (1)Muyang Li (1)

Papers (1)

Mar 10, 2026

BAIR6d ago·also MIT CSAIL, NVIDIA, Tsinghua AI

Flash-KMeans: Fast and Memory-Efficient Exact K-Means

K-means, previously relegated to offline processing, gets a 17.9x speed boost on modern GPUs thanks to Flash-KMeans' clever IO and contention optimizations.

Shuo Yang, Haocheng Xi, Yilong Zhao +10

Distributed Systems & Hardware Inference & Quantization Training Efficiency & Optimization

Search

Yujun Lin

Publication activitypapers/week, last 8 weeks

Research focus

Frequent co-authors

Papers (1)