Arjun Menon
AI Researcher at Princeton University
Hi there! I’m Menon, a senior at Princeton University studying Computer Science.
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…)
- Bayesian 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 -> Sampura Research
- Mechanistic Interpretability with the Princeton Natural and Artificial Minds Initiative (see here)
- 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
More details can be found on my CV.
Research Interests
My work focuses on embodied reasoning, primarily with respect to robotics foundation models. I utilize my joint priors from cognitive science and systems to work on efficiently replicating human-like levels of abstract reasoning on non-human hardware. Within that scope, I work on a wide range of problems, including but not limited to robotics-specific model architecture development, efficient training/inference, encoding cognitive priors, constraining embedding geometry, and guiding learned representations.
Publications
You can find my past projects and research here. Some highlights include…
- CODA: Rewriting Transformer Blocks as GEMM-Epilogue Programs (NeurIPS 2026 Poster; Cited by NVIDIA, ByteDance)
- Exploring the Effectiveness of Geometrically Structured Representations for Visual Imitation
- Evidence Against LLM Homogenization in Creative Writing (Cited in 2 books)
- Towards Trustworthy Physical AI: From Theory to Practice Across Life Cycle (Project Advised by Yoshua Bengio)
- Toward Human-AI Complementarity Across Diverse Tasks
- 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, student government, poetry, cooking/baking, and a few other things…