Principal Research Engineer - Generative AI - AI Frontiers

Microsoft Microsoft · Big Tech · Redmond, WA +1 · Applied Sciences

Research Engineer at Microsoft Research's AI Frontiers lab focused on advancing AI capabilities in modeling, algorithms, reasoning, and agentic AI. The role involves developing novel ideas in reinforcement learning, evolving pre-training and post-training codebases, and collaborating on state-of-the-art reasoning and agentic AI. Opportunities to ship research to Microsoft customers and expand AI capabilities, efficiency, and safety.

What you'd actually do

  1. As a Principal Research Engineer in AI Frontiers, you will design, develop, execute, and implement technology research projects in collaboration with other researchers, engineers, and product groups.
  2. 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.
  3. Embody our culture and values.

Skills

Required

  • 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.

Nice to have

  • 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.

What the JD emphasized

  • advancing AI capabilities
  • reinforcement learning research
  • pre-training
  • mid-training
  • post-training
  • reasoning
  • agentic AI
  • foundation models
  • learning agent platforms

Other signals

  • advancing AI capabilities
  • reinforcement learning research
  • pre-training, mid-training, and post-training codebases
  • reasoning and agentic AI
  • foundation models and learning agent platforms