avatar

Nived Rajaraman

PhD Student
UC Berkeley
nived.rajaraman@berkeley.edu


About Me

I am a postdoc at Microsoft Research NYC in the Reinforcement Learning group. I recently finished my PhD at UC Berkeley, advised by Jiantao Jiao and Kannan Ramchandran, affiliated with the BLISS and BAIR labs. I was previously an intern with Nevena Lazic and Dong Yin at Deepmind and with Ravishankar Krishnaswamy at MSR. While at Berkeley, I organized the BLISS and CLIMB seminars.

I work on a variety of topics in the theory and practice of machine learning with a general focus on the statistical and computational aspects of adaptive decision making problems and RL. More recently, I have been interested in the application of these techniques in pushing our understanding of large language models. My research has largely focused on using mathematical frameworks to provide intuitions missing in existing empirical approaches.

In a previous life, I was a dual degree student at the Department of Electrical Engineering, IIT Madras. I am fortunate to have had Andrew Thangaraj as my thesis advisor and to have worked closely with Rahul Vaze.

Updates

May 2025: Organizing a COLT ‘25 workshop on the Foundations of Post-training. Submission deadline: May 23, 2025!

Jan 2025: I am a CPAL Rising Star for the year 2025.

July 2023: I passed my qualifying exam! Thanks to my committee members, Profs. Jiantao Jiao, Kannan Ramchandran, Sasha Rakhlin, and chaired by Prof. Mike Jordan.

Publications

  1. From Markov to Laplace: How Mamba In-context Learns Markov chains
    M. Bondaschi, N. Rajaraman, M. Gastpar, K. Ramchandran, C. Gulcehre, R. Pascanu, A. V. Makkuva
    ICLR 2026 (oral)
  2. A. Setlur, N. Rajaraman, S. Levine, A. Kumar
    ICML 2025 (spotlight)
  3. SPECS: Faster Test-Time Scaling through Speculative Drafts
    M. Cemri, N. Rajaraman, R. Tiwari, X. Liu, K. Keutzer, I. Stoica, K. Ramchandran, A. Beirami, Z. Sun
    ICML 2025 Workshop on Efficient Systems for Foundation Models (ES-FOMO III)
  4. A. Assosα, Y. Daganα, N. Rajaramanα
    COLT 2025
  5. The Space Complexity of Learning Unlearning Algorithms
    Y. Cherapanamjeri, S. Garg, N. Rajaraman, A. Sekhari, A. Shetty
    COLT 2025
  6. N. Rajaraman, J. Jiao, K. Ramchandran
    Neurips 2024 (spotlight)
  7. N. Rajaraman, M. Bondaschi, A. V. Makkuva, K. Ramchandran, M. Gastpar
    Neurips 2024; ICML 2024 Workshop on Mechanistic Interpretability 2024
  8. N. Rajaraman, Y. Han, J. Jiao, K. Ramchandran
    Annals of Statistics
  9. N. Rajaraman, Devvrit, A. Mokhtari, K. Ramchandran
    NeurIPS 2023
  10. D. Yin, S. Thiagarajan, N. Lazic, N. Rajaraman, B. Hao, C. Szepesvari
  11. G. Swamy*, N. Rajaraman*, M. Peng, S. Choudhury, J. Bagnell, S. Z. Wu, J. Jiao, K. Ramchandran
    NeurIPS 2022
  12. N. Rajaraman, Devvrit, P. Awasthi
    NeurIPS 2022
  13. A. Aghazadeh, N. Rajaraman, T. Tu, K. Ramchandran
    TMLR
  14. N. Rajaraman, Y. Han, L. Yang, J. Liu, J. Jiao, K. Ramchandran
    NeurIPS 2021
  15. N. Rajaraman, Y. Han, L. F. Yang, K. Ramchandran, J. Jiao
    Submitted to the Annals of Statistics
  16. N. Rajaraman, L. F. Yang, J. Jiao, K. Ramchandran
    NeurIPS 2020
  17. S. Kadhe, N. Rajaraman, O. O. Koyluoglu, K. Ramchandran
    ICML Workshop on FL for User Privacy and Data Confidentiality (2020); CCS Workshop on Privacy-Preserving Machine Learning in Practice (2020); ISIT (2021)
  18. R. Krishnaswamyα, Devvrit* α, N. Rajaraman*α
    APPROX/RANDOM 2019

Powered by Jekyll and Minimal Light theme.