Microsoft is looking to solve complex challenges in diverse fields, including computing, healthcare, economics, and the environment, by developing innovative techniques for automating and optimizing computational kernels to accelerate modern AI systems.
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
- 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
- Contribute to the forefront of AI and high-performance computing
- Develop innovative techniques for automating and optimizing computational kernels
- Engage in pushing the capabilities of AI models for kernel rewriting and fusion across GPU, MAIA, and AMD platforms
- Rigorously benchmark new solutions against real-world workloads
- Work with algorithms, compilers, and machine learning
- Make impactful contributions to open-source projects and academic literature
- Collaborate with other Research Interns and researchers, present findings, and contribute to the vibrant life of the community
Other
- Enrolled in a PhD program
- Physically located in their manager’s Microsoft worksite location for the duration of their internship
- Submit a minimum of two reference letters for this position as well as a cover letter and any relevant work or research samples
- Be ready to submit your letter writers in advance
- Comply with Microsoft's equal opportunity employment policies
