Research Engineer - Generative AI - AI Frontiers

Microsoft is seeking a Research Engineer to join their AI Frontiers lab, which focuses on advancing Artificial Intelligence capabilities. The role involves designing, developing, and executing technology research projects in collaboration with other researchers and engineers, particularly in the areas of Large Language Models and agentic AI.

Responsibilities

• As a Research Engineer in AI Frontiers, you will design, develop, execute, and implement technology research projects in collaboration with other researchers, engineers, and product groups
• As a member of a world-class research organization, you will be a part of research breakthroughs in the field and play a crucial role in developing, improving, and exploring the capabilities of Large Language Models (LLMs), reasoning and agentic AI
• Embody our culture and values

Skills

• Bachelors in Computer Science or relevant field AND 2+ years related experience
• OR Master's Degree in Computer Science or related field AND 1+ year(s) related experience
• OR Doctorate in Computer Science or related field 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
• Familiarity 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

Company Overview

Company H1B Sponsorship