MONGOOSE: A Learnable LSH Framework for Efficient Neural Network Training. - Beidi Chen; Zichang Liu; Binghui Peng; Zhaozhuo Xu; Jonathan Lingjie Li; Tri Dao; Zhao Song 002; Anshumali Shrivastava; Christopher Ré (ICLR, 2021)

Training Domain Draft Models for Speculative Decoding: Best Practices and Insights. - Fenglu Hong; Ravi Raju; Jonathan Lingjie Li; Bo Li; Urmish Thakker; Avinash Ravichandran; Swayambhoo Jain; Changran Hu (CoRR, 2025)