Transformers to SSMs: Distilling Quadratic Knowledge to Subquadratic Models. - Aviv Bick; Kevin Y. Li; Eric P. Xing; J. Zico Kolter; Albert Gu (NeurIPS, 2024)

Inference Optimal VLMs Need Only One Visual Token but Larger Models. - Kevin Y. Li; Sachin Goyal; João D. Semedo; J. Zico Kolter (CoRR, 2024)

Inference Optimal VLMs Need Fewer Visual Tokens and More Parameters. - Kevin Y. Li; Sachin Goyal; João D. Semedo; J. Zico Kolter (ICLR, 2025)

Thinking Slow, Fast: Scaling Inference Compute with Distilled Reasoners. - Daniele Paliotta; Junxiong Wang; Matteo Pagliardini; Kevin Y. Li; Aviv Bick; J. Zico Kolter; Albert Gu; François Fleuret; Tri Dao (CoRR, 2025)