Senior Machine Learning Engineer

Lead the training of large language and multimodal models, including developing training workflows as well as managing, optimizing, and debugging such runs. Develop data and evaluation pipelines designed to maximize the performance of trained models. Mentor and provide technical leadership across the team and org 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. Bachelors or Masters in Computer Science or related technical discipline and 3+ years of technical engineering experience in ML-related work OR a Doctorate in Computer Science or related technical discipline. Applied technical experience in one of: distributed training of deep learning models, data pipelines for model training, or evaluation workflows for assessment of model capabilities.

Deep experience with PyTorch or Jax. Experience with 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: optimizing data mixes, midtraining, post-training, model merging, or model distillation. Effective communication skills with the ability to present complex technical concepts to a broad audience

Microsoft