Nicholas Lee

I work on AI agents, efficient machine learning systems, and spoken language understanding.

Nicholas Lee

Education

Ph.D.
Computer Science, UC Berkeley Co-advised by Kurt Keutzer and Gopala Anumanchipalli 2023–Present
M.S.
5th Year M.S. in Computer Science, UC Berkeley 2022–2023
B.S.
Electrical Engineering and Computer Sciences, UC Berkeley 2018–2022

Selected publications

Full list on Google Scholar
  1. 2026

    Agentic Test-Time Scaling for WebAgents

    Nicholas Lee*, Lutfi Eren Erdogan*, Chris Joseph John, Surya Krishnapillai, Michael W. Mahoney, Kurt Keutzer, Amir Gholami

    COLM Workshop on Efficient Reasoning

  2. 2026

    StyleStream: Real-Time Zero-Shot Voice Style Conversion

    Yisi Liu, Nicholas Lee, Gopala K. Anumanchipalli

    Interspeech, Long Paper Track

  3. 2025

    Plan-and-Act: Improving Planning of Agents for Long-Horizon Tasks

    Lutfi Eren Erdogan*, Nicholas Lee*, Sehoon Kim, Suhong Moon, Hiroki Furuta, Gopala Anumanchipalli, Kurt Keutzer, Amir Gholami

    International Conference on Machine Learning (ICML)

  4. 2025

    Sylber: Syllabic Embedding Representation of Speech from Raw Audio

    Cheol Jun Cho, Nicholas Lee, Akshat Gupta, Dhruv Agarwal, Ethan Chen, Alan W. Black, Gopala K. Anumanchipalli

    International Conference on Learning Representations (ICLR)

  5. 2024

    LLM2LLM: Boosting LLMs with Novel Iterative Data Enhancement

    Nicholas Lee*, Thanakul Wattanawong*, Sehoon Kim, Karttikeya Mangalam, Sheng Shen, Gopala Anumanchipalli, Michael W. Mahoney, Kurt Keutzer, Amir Gholami

    Findings of ACL

  6. 2024

    TinyAgent: Function Calling at the Edge

    Lutfi Eren Erdogan*, Nicholas Lee*, Siddharth Jha*, Sehoon Kim, Ryan Tabrizi, Suhong Moon, Coleman Hooper, Gopala Anumanchipalli, Kurt Keutzer, Amir Gholami

    Empirical Methods in Natural Language Processing (EMNLP)

  7. 2024

    An LLM Compiler for Parallel Function Calling

    Sehoon Kim*, Suhong Moon*, Ryan Tabrizi, Nicholas Lee, Michael W. Mahoney, Kurt Keutzer, Amir Gholami

    International Conference on Machine Learning (ICML)

  8. 2023

    S-LoRA: Serving Thousands of Concurrent LoRA Adapters

    Ying Sheng*, Shiyi Cao*, Dacheng Li, Coleman Hooper, Nicholas Lee, Shuo Yang, Christopher Chou, Banghua Zhu, Lianmin Zheng, Kurt Keutzer, Joseph E. Gonzalez, Ion Stoica

    Conference on Machine Learning and Systems (MLSys)

  9. 2022

    Squeezeformer: An Efficient Transformer for Automatic Speech Recognition

    Sehoon Kim*, Amir Gholami*, Albert Shaw†, Nicholas Lee†, Karttikeya Mangalam, Jitendra Malik, Michael W. Mahoney, Kurt Keutzer

    Advances in Neural Information Processing Systems (NeurIPS)

* Equal contribution. † Equal second-author contribution on Squeezeformer.

Teaching

Fall 2026

CS 183/283: Natural Language Processing