Principal Research Engineer – Generative AI – AI Frontiers

Job Description

Principal Research Engineer position at Microsoft Research AI Frontiers lab focusing on advancing Artificial Intelligence capabilities through cutting-edge research and collaboration.

Requirements

• Bachelors in Computer Science or relevant field AND 6+ years related experience OR Master's Degree in Computer Science or related field AND 4+ years related experience OR Doctorate in Computer Science or related field AND 3+ years related experience OR equivalent experience
• 1+ year(s) experience developing with Python and Pytorch/JAX
• Familiarity with architecture and optimizations for large language models
• Hands-on work in debugging and profiling Pytorch distributed code
• Basic understanding of working of CUDA kernels
• Experience with pre-training, mid-training, and/or post-training pipelines for language and/or multimodal models
• Foundational understanding of reinforcement learning and key challenges in the field
• Experience with verl, Ray, Megatron and/or vLLM is a significant plus
• Any experience in building scalable services can be highly complementary

Responsibilities

• Design, develop, execute, and implement technology research projects in collaboration with other researchers, engineers, and product groups
• Contribute to research breakthroughs in Large Language Models (LLMs), reasoning, and agentic AI
• Play a crucial role in developing, improving, and exploring the capabilities of AI technologies

Benefits

• Industry leading healthcare
• Educational resources
• Discounts on products and services
• Savings and investments
• Maternity and paternity leave
• Generous time away
• Giving programs
• Opportunities to network and connect