Arjun Parthasarathy

Undergraduate ML Researcher at Columbia University

prof_pic.jpeg

Columbia University

New York, NY 10027

Hi! I’m an undergraduate at Columbia University studying Computer Science and Applied Math. I previously interned at Jump Trading, where I worked on GPU kernel optimization for training. I’m grateful to be supported by the I.I. Rabi Scholarship.

selected publications

  1. world_model_headline.png
    Closing the Train-Test Gap in World Models for Gradient-Based Planning
    Rohun Agrawal, Nimit Kalra, Arjun Parthasarathy, Yann LeCun, Oumayma Bounou, Pavel Izmailov, and Micah Goldblum
    arXiv preprint arXiv:2512.09929, 2025
    arXiv PDF
  2. tk_headline.png
    ThunderKittens: Simple, Fast, and Adorable Kernels
    Benjamin Frederick Spector, Simran Arora, Aaryan Singhal, Arjun Parthasarathy, Daniel Y Fu, and Christopher Re
    In The Thirteenth International Conference on Learning Representations, 2025
    OpenReview PDF
  3. defer_headline.png
    DEFER: Distributed Edge Inference for Deep Neural Networks
    Arjun Parthasarathy, and Bhaskar Krishnamachari
    In 2022 14th International Conference on COMmunication Systems & NETworkS (COMSNETS) . Best Paper Award , 2022
    arXiv PDF
  4. seifer_headline.png
    SEIFER: Scalable Edge Inference for Deep Neural Networks
    Arjun Parthasarathy, and Bhaskar Krishnamachari
    Workshop on Challenges in Deploy and Monitoring ML Systems, Conference on Neural Information Processing Systems (NeurIPS), 2022
    arXiv PDF
  5. itnac_headline.png
    Partitioning and Placement of Deep Neural Networks on Distributed Edge Devices to Maximize Inference Throughput
    Arjun Parthasarathy, and Bhaskar Krishnamachari
    In 2022 32nd International Telecommunication Networks and Applications Conference (ITNAC), 2022
    arXiv PDF