Cambridge Research Internship on Compiler-like Tools for LLM Efficiency

Job Description

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. Consistent with applicable law, 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. If you need assistance and/or a reasonable accommodation due to a disability during the application process, read more about requesting accommodations.