New Loss Functions for Fast Maximum Inner Product Search. - Ruiqi Guo; Quan Geng; David Simcha; Felix Chern; Sanjiv Kumar; Xiang Wu (CoRR, 2019)

Accelerating Large-Scale Inference with Anisotropic Vector Quantization. - Ruiqi Guo; Philip Sun; Erik Lindgren; Quan Geng; David Simcha; Felix Chern; Sanjiv Kumar (ICML, 2020)

TPU-KNN: K Nearest Neighbor Search at Peak FLOP/s. - Felix Chern; Blake Hechtman; Andy Davis; Ruiqi Guo; David Majnemer; Sanjiv Kumar (NeurIPS, 2022)

Large Models are Parsimonious Learners: Activation Sparsity in Trained Transformers. - Zonglin Li; Chong You; Srinadh Bhojanapalli; Daliang Li; Ankit Singh Rawat; Sashank J. Reddi; Ke Ye; Felix Chern; Felix X. Yu; Ruiqi Guo; Sanjiv Kumar (CoRR, 2022)

The Lazy Neuron Phenomenon: On Emergence of Activation Sparsity in Transformers. - Zonglin Li; Chong You; Srinadh Bhojanapalli; Daliang Li; Ankit Singh Rawat; Sashank J. Reddi; Ke Ye; Felix Chern; Felix X. Yu; Ruiqi Guo; Sanjiv Kumar (ICLR, 2023)

Spark Transformer: Reactivating Sparsity in FFN and Attention. - Chong You; Kan Wu; Zhipeng Jia; Lin Chen; Srinadh Bhojanapalli; Jiaxian Guo; Utku Evci; Jan Wassenberg; Praneeth Netrapalli; Jeremiah J. Willcock; Suvinay Subramanian; Felix Chern; et al. (CoRR, 2025)

EQuARX: Efficient Quantized AllReduce in XLA for Distributed Machine Learning Acceleration. - Ibrahim Ahmed 012; Clemens Schaefer; Gil Tabak; Denis Vnukov; Zenong Zhang; Felix Chern; Anatoliy Yevtushenko; Andy Davis (CoRR, 2025)