Senior Machine Learning Engineer

Microsoft•Cambridge, MA

9h•$119,800 - $261,000

About The Position

We are seeking a Senior Machine Learning Engineer to design, train, and optimize generative models for multimodal scientific discovery. This is a hands-on technical role where you’ll work alongside a multi-disciplinary team of ML researchers, engineers, and domain experts to develop the methods and tools needed to accelerate the rate of discovery and further the frontier of human knowledge. There is no one-size-fits-all approach – advancing model capabilities requires investment throughout the training lifecycle, including method development, recipe refinement, scaling and throughput optimization, and model-harness co-design. In this role, you’ll contribute to these workstreams and leverage their learnings to drive scaled runs. Microsoft Research New England is comprised of full-time researchers, postdocs, data scientists, engineers, and program managers working in machine learning, statistics, healthcare, computational biology, theoretical computer science, algorithmic game theory, economics, among other disciplines. The lab is highly collaborative and interdisciplinary and is actively engaged with the local academic and tech communities. We are committed to building an inclusive, diverse, and pluralistic research environment. In keeping with this, Microsoft Research New England is a place where all are able to do their very best work. Microsoft’s mission is to empower every person and every organization on the planet to achieve more. As employees we come together with a growth mindset, innovate to empower others, and collaborate to realize our shared goals. Each day we build on our values of respect, integrity, and accountability to create a culture of inclusion where everyone can thrive at work and beyond.

Requirements

Bachelor's Degree in Computer Science or related technical field AND 4+ years technical engineering experience in ML-related work OR equivalent experience.
Hands-on technical experience in one of: distributed training of large language / multimodal models (pretraining, midtraining, or post-training), optimizing training workflows, or model distillation
Deep experience with PyTorch or Jax.
Experience designing, optimizing, and monitoring generative modeling training workflows.
Familiarity with profiling training and/or inference code, identifying performance bottlenecks, and mitigating them.
Comfort with one or more of: refining data mixes, midtraining, post-training, data/model parallelism techniques, optimizing training runs, or model distillation.
Demonstrated communication skills with the ability to present complex technical concepts to a broad audience

Nice To Haves

Bachelor's Degree in Computer Science or related technical field AND 4+ years technical engineering experience in ML-related work OR a Doctorate in Computer Science or related technical field.

Responsibilities

Lead the training of large language and multimodal models, including developing, managing, and debugging training workflows
Scale and optimize training runs, identifying opportunities for efficiency gains and implementing them
Mentor and provide technical leadership across the team and organization in your area(s) of expertise.
Explore and advise on strategic direction and technical approaches for projects across the team.
Collaborate with researchers, domain experts, and cross-disciplinary teams to ensure our techniques translate the latest research into impactful solutions.