Research Intern - AI Compiler
About The Position
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. This Research Internship presents a unique opportunity to contribute to the forefront of AI and high-performance computing. The project centers on developing innovative techniques for automating and optimizing computational kernels, which are essential for accelerating modern AI systems. Research Interns will engage in pushing the capabilities of AI models for kernel rewriting and fusion across Graphical Processing Unit (GPU), Microsoft AI Accelerator (MAIA), and Applied Micro Devices (AMD) platforms, and rigorously benchmark new solutions against real-world workloads. The research experience will include working with algorithms, compilers, and machine learning, and making impactful contributions to open-source projects and academic literature. Join the team and help push the boundaries of AI for AI systems. 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.
Requirements
- Enrolled in a PhD program in Machine Learning/AI/Compiler/Algorithms/Systems or related area.
- 1+ years of experience with deep learning frameworks like PyTorch.
- 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.
Nice To Haves
- Experience with AI compilers.
- Experience with post-training models: Supervised Fine-Tuning, Reinforcement Learning, etc.
- Experience with writing GPU kernels in Triton/Compute Unified Device Architecture/Gluon/Helion etc.
- Demonstrated ability to develop original research agendas.
Responsibilities
- 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.
Stand Out From the Crowd
Upload your resume and get instant feedback on how well it matches this job.
What This Job Offers
Career Level
Intern
Education Level
Ph.D. or professional degree
Number of Employees
5,001-10,000 employees
