Ion Stoica

Papers on Lattice

Total citations

Topics

h-index

Publication activitypapers/week, last 8 weeks

Research focus

Training Efficiency & Optimization (2)Distributed Systems & Hardware (1)Inference & Quantization (1)Architecture Design (Transformers, SSMs, MoE) (1)

Frequent co-authors

Shuo Yang (2)Haocheng Xi (2)Jintao Zhang (2)Kurt Keutzer (2)

Papers (2)

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

Mar 9, 2026

BAIR1w ago·also Tsinghua AI

SVG-EAR: Parameter-Free Linear Compensation for Sparse Video Generation via Error-aware Routing

Get 2x faster video generation from diffusion transformers without sacrificing quality, thanks to a clever parameter-free error compensation technique.

Xuanyi Zhou, Qiuyang Mang, Shuo Yang +7

Architecture Design (Transformers, SSMs, MoE)Computer Vision Training Efficiency & Optimization

Search

Ion Stoica

Publication activitypapers/week, last 8 weeks

Research focus

Frequent co-authors

Papers (2)