Arjun Menon
AI Researcher at Princeton University
Reach out at armenon [at] princeton (dot) edu
Hi there! I’m Menon, a junior at Princeton University studying CS and Math.
I currently work on…
- Robotic Intelligence in the Princeton Robotic Intelligence and Systems Lab under the advisory of Dhruv Shah @ Princeton (stay tuned for upcoming publications…)
- Reinforcement Learning in the Computational Cognitive Science Lab under the advisory of Tom Griffiths and Dilip Arumugam @ Princeton (stay tuned for upcoming publications…)
- LLM Kernel-Writing Augmentation in the Dao AI Lab under the advisory of Tri Dao @ Together AI/Princeton and Han Guo @ MIT (see latest)
I have previously worked on...
- Human-AI Complementarity through SPAR under the advisory of Rishub Jain @ Google DeepMind
- Mechanistic Interpretability with the Princeton Natural and Artificial Minds Initiative
- Optimization for asset generation in Infinigen with the Princeton Vision & Learning Lab
and dabbled in industry at...
- Artemis Ops as an LLM and Backend Software Engineering Intern
- Palantir’s Winter Defense Tech Research Fellowship
Research Interests
I’m currently most interested in reinforcement learning and robotics foundation models, but also dabble in reasoning and neuromimetic learning algorithms.
Publications
You can find my past projects and research here. Some highlights include…
- CODA: Rewriting Transformer Blocks as GEMM-Epilogue Programs
- Toward Human-AI Complementarity Across Diverse Tasks
- Evidence Against LLM Homogenization in Creative Writing
- RotoSLAM: Improving SLAM Scale Consistency for Rotational Motion
Awards & Honors
3rd Overall Anthropic Invitational Alignment Hackathon 2025
2nd Overall UChicago Trading Competitions 2024
Princeton Top U-20 Scorer Putnam 2023
Other Interests
I dabble in photography, theatre, poetry, cooking/baking, and a few other things…