Yash Jain

Welcome! I am a Research Scientist at Essential AI Labs, where I build foundation models alongside Ashish Vaswani (Attention is All You Need). I was a core contributor to Rnj-1, a state-of-the-art open-source coding and agentic foundation model that has amassed 600k+ downloads.

Previously, I was an ML Scientist II at Microsoft Research, where I worked on diffusion models and multimodal large-language models in collaboration with Vibhav Vineet, publishing at CVPR, NeurIPS, NAACL, and EMNLP.

I graduated from Georgia Tech with an M.S. in Computer Science, advised by Zsolt Kira. Before that, I earned my B.Tech. in Computer Science from IIT Bombay, where I received an Excellence in Research Award under Soumen Chakrabarti.

Reach out by email if you wish to collaborate!