The Future AI Infrastructure team at Microsoft Research Cambridge (UK) is dedicated to developing technologies for AI data centers of the future. We are seeking a masters/PhD student to join us in Cambridge winter/spring 2025 to work on automatic, compiler-like tools for LLM efficiency, covering topics such as quantization and parallelization. You will joining a welcoming and highly collaborative environment and work on creative and challenging problems during your internship. 

Responsibilities

• Research and develop novel compiler-like tools for LLM efficiency 

• Design, implement and evaluate such tools 

• Write and present your findings in technical documents or presentations 

Qualifications

Required/Minimum Qualifications:   

• Be enrolled in Masters/PhD program in Computer Science/Machine Learning or related discipline 

• Substantial experience across static analysis, verification, constraint solving  

• Substantial experience across LMs, inference, parallelization, quantization 

Other Requirements:  

• Interns are expected to be physically located in Cambridge, UK 

Preferred/Additional Qualifications: 

• Strong technical/scientific communication 

• PyTorch, CUDA, tooling 

Microsoft is an equal opportunity employer. All qualified applicants will receive consideration for employment without regard to age, ancestry, color, family or medical care leave, gender identity or expression, genetic information, marital status, medical condition, national origin, physical or mental disability, political affiliation, protected veteran status, race, religion, sex (including pregnancy), sexual orientation, or any other characteristic protected by applicable laws, regulations and ordinances. If you need assistance and/or a reasonable accommodation due to a disability during the application or the recruiting process, please send a request via the Accommodation request form.

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