Applied Research Scientist - AI Models & Agents

About the role

The AI Models team is looking for exceptional machine learning scientists and engineers to explore and innovate on training and inference techniques for large language models (LLMs), large multimodal models (LMMs), image/video generation and other foundation models as well as self-evolving agents on top of these.

What you'll do

• Be part of a world-class research and development team focussing on efficient and scalable pre-training, instruction tuning, alignment and optimization
• Build AI agents to automatically write and evaluate efficient kernels and code Drive continuous improvement of infrastructure and development ecosystem Publish your…
• Engage with academia and open-source ML communities

What you need

• Research
• Machine learning
• LLMs
• Deep learning
• PyTorch
• TensorFlow
• PhD or B.Tech or B.E. or B.Sc or BCA or M.Tech or MCA

Good to have

• Python