Cambridge Residency Programme – AI Parallelisation

Key Skills

Aartificial intelligence, C++, Deep Learning, Design, Large Language Models - LLMs, Machine learning techniques, NeurIPS, Python Programming

Job Description

We are looking for a Postdoctoral researcher to join us as part of a two-year AI Residency Program in Cambridge, UK. This is a close collaboration between M365 Research and The Cloud Infrastructure team in Microsoft Research to advance the state of the art in AI efficiency. 

Generative AI is revolutionizing how we create, share, work with, and consume content. At Microsoft, we run the biggest platform for collaboration and productivity in the world with hundreds of millions of consumer/enterprise users. Tackling AI efficiency challenges is crucial for delivering these experiences at scale.  

Within our Microsoft wide Systems Innovation initiative, we are working to advance efficiency across AI systems, where we look at novel designs and optimizations across AI stacks: models, AI frameworks, cloud infrastructure, and hardware.  

We believe that dramatic improvements to cloud-scale AI/ML systems can be enabled by automatic tools supporting their development and deployment process. This includes architectural exploration, mapping of workloads onto the infrastructure, and further optimizations. There is a wealth of research opportunities and challenges in scheduling and optimization, constraint solving, program synthesis, local search, and other areas. We are building such tools, in collaboration with product groups, other teams at Microsoft Research, and the broader academic community

Qualifications

Required/Minimum Qualifications:  

• PhD in computer science, math, physics, or related field
• Hands on experience with constraint solving, automated reasoning, program verification, DNNs/LLMs, or related. 
• Capability for rapid and accurate development in, for example, Python, C++, or similar.  

Preferred/Additional Qualifications:

• Publishing track record in automation of distributed deep learning, constraint solving, program verification (e.g., NeurIPS, ICML, OSDI, CP, SAT, CAV, PLDI, POPL).
• Experience with ML frameworks.

Responsibilities

You will work closely with other researchers to advance research on automated tools for the design, development, and deployment of AI/ML infrastructure and workloads at scale. This includes but is not limited to algorithms for scheduling communication over heterogeneous networks, partitioning/fusion of computation tasks, and static analysis of DNNs. You will have access to our internal product partners and will closely work with experts on AI/ML systems, automated reasoning, and constraint solving. Furthermore, you will contribute to the research community by publishing and presenting your work at scientific conferences.

Benefits/perks listed below may vary depending on the nature of your employment with Microsoft and the country where you work.

• Industry leading healthcare
• Educational resources
• Discounts on products and services
• Savings and investments
• Maternity and paternity leave
• Generous time away
• Giving programs
• Opportunities to network and connect