Research Intern - LLM Quantization

Research Internships at Microsoft provide a dynamic environment for research careers with a network of world-class research labs led by globally-recognized scientists and engineers, who pursue innovation in a range of scientific and technical disciplines to help solve complex challenges in diverse fields, including computing, healthcare, economics, and the environment.

Our team works on performance analysis and optimization of large language models, spanning the stack from GPU kernel implementation through to changes in model architecture. A key challenge is that quantization of models to use smaller data types is only effective if we can dequantize the formats and use them efficiently during computation. In this Research Internship we are going to tackle this problem by exploring the co-design of quantization techniques (e.g., fewer bits per weight) and kernel design for efficient decode (e.g., expanding weights to the 4-bit, 6-bit and 8-bit floating point formats in modern GPUs).

Required Qualifications

• Accepted or currently enrolled in a PhD program in Computer Science or related STEM field.
• At least 1 year of experience working with GPU-based algorithms.

Other Requirements

• Research Interns are expected to be physically located in their manager’s Microsoft worksite location for the duration of their internship.
• In addition to the qualifications below, you’ll need to submit a minimum of two reference letters for this position as well as a cover letter and any relevant work or research samples. After you submit your application, a request for letters may be sent to your list of references on your behalf. Note that reference letters cannot be requested until after you have submitted your application, and furthermore, that they might not be automatically requested for all candidates. You may wish to alert your letter writers in advance, so they will be ready to submit your letter. 

Preferred Qualifications

• Expertise with performance analysis and profiling.
• Hands-on CUDA or ROCm GPU kernel development is preferred.

The base pay range for this internship is USD $6,550 - $12,880 per month. There is a different range applicable to specific work locations, within the San Francisco Bay area and New York City metropolitan area, and the base pay range for this role in those locations is USD $8,480 - $13,920 per month.

Certain roles may be eligible for benefits and other compensation. Find additional benefits and pay information here: https://careers.microsoft.com/us/en/us-intern-pay 

Microsoft accepts applications and processes offers for these roles on an ongoing basis.

Research Interns put inquiry and theory into practice. Alongside fellow doctoral candidates and some of the world’s best researchers, Research Interns learn, collaborate, and network for life. Research Interns not only advance their own careers, but they also contribute to exciting research and development strides. During the 12-week internship, Research Interns are paired with mentors and expected to collaborate with other Research Interns and researchers, present findings, and contribute to the vibrant life of the community. Research internships are available in all areas of research, and are offered year-round, though they typically begin in the summer.