Senior Research Engineer Machine Learning,AI for Science in Cambridge

Overview

The mission of the team is to enable highly accurate and scalable predictions of the electronic structure of molecules and materials using deep learning-powered DFT. Accurate predictions of electronic structure are essential for predictive modeling across chemistry and materials science.

The primary focus of this role will be developing and maintaining efficient, reliable end-to-end infrastructure for large-scale model training, evaluation, and benchmarking, including large-scale data pipelines for heterogeneous scientific data, distributed training, experimentation, and agentic workflows. If you have solid machine learning engineering expertise and enjoy building scalable infrastructure that enables cutting-edge machine learning research, we encourage you to apply. This post will be open until the positions are filled.

Responsibilities

• Develop and maintain efficient, reliable, and scalable infrastructure for large-scale machine learning training, evaluation, and benchmarking.
• Architect, design, and implement large-scale data pipelines for heterogeneous scientific data, supporting efficient model training, experimentation, evaluation, and benchmarking.
• Build and optimize distributed data processing, model training, evaluation, and benchmarking pipelines.
• Develop software and agentic workflows to automate and improve model development, training, evaluation, and benchmarking.
• Continuous monitoring and improvement of the stability of data, training, evaluation and benchmarking workflows.
• Work cross-functionally with machine learning researchers, engineers, and researchers from the natural sciences.
• Maintain high standards in code quality and software design.
• Document and share practices across the organization.

Qualifications

Required qualifications:

• Completed MSc in computer science, machine learning, AI or a related area.
• Proficiency in collaborative software engineering in Python.
• In-depth understanding of machine learning frameworks such as PyTorch and/or Jax.
• Experience in designing, developing and deploying ML systems.
• Experience with agentic software development.
• Experience building and optimizing distributed systems and large-data applications, including those using tensor accelerators or GPUs.
• Ability to work in an interdisciplinary collaborative environment, through effective communication of technical concepts to non-experts from different technical backgrounds.

Qualifications that are considered a plus, but are not strictly required:

• PhD degree in computer science, machine learning, AI or a related field, or comparable industry experience in working with machine learning and large datasets.
• Experience working with major cloud platforms and/or HPC.
• Experience developing high-performance scientific software.
• Experience designing or developing agentic frameworks, orchestration systems, or infrastructure for building and running AI agents.

This position will be open for a minimum of 5 days, with applications accepted on an ongoing basis until the position is filled. Microsoft is an equal opportunity employer. All qualified applicants will receive consideration for employment without regard to age, ancestry, citizenship, color, family or medical care leave, gender identity or expression, genetic information, immigration status, marital status, medical condition, national origin, physical or mental disability, political affiliation, protected veteran or military status, race, ethnicity, religion, sex (including pregnancy), sexual orientation, or any other characteristic protected by applicable local laws, regulations and ordinances.