Senior Researcher - GPU Performance
Microsoft
Senior Researcher – GPU Performance
Bangalore, Karnataka, India
Save
Overview
Within our Microsoft wide Systems Innovation initiative, we are working to advance efficiency across AI systems, where we look at novel designs and optimizations across AI stacks: models, AI frameworks, cloud infrastructure, and hardware. We are an Applied Research team driving mid- and long-term product innovations. We closely collaborate with multiple research teams and product groups across the globe who bring a multitude of technical expertise in cloud systems, machine learning and software engineering. We communicate our research both internally and externally through academic publications, open-source releases, blog posts, patents, and industry conferences. Further, we also collaborate with academic and industry partners to advance the state of the art and target material product impact that will affect 100s of millions of customers.
We are looking for a Senior Researcher – Hardware/Software Codesign researcher to explore hardware/kernel-level optimizations to deliver significant efficiency gains for Large Language Models and Generative AI experiences.
The ideal candidate will have a strong background in GPU architecture, accelerator design, machine learning, or systems research and the ambition to apply them to large scale production systems. This role combines deep technical expertise in GPU architecture with practical implementation skills to create efficient, scalable computational kernels. Further, the ideal candidate must have demonstrated a history of solving hard technical problems and is motivated to tackle the hardest problems in building a full end-to-end AI stack. An entrepreneurial approach and ability to take initiative and move fast are essential.
Have a look at this link for reading: Efficient AI - Microsoft Research
Microsoft’s mission is to empower every person and every organization on the planet to achieve more. As employees we come together with a growth mindset, innovate to empower others, and collaborate to realize our shared goals. Each day we build on our values of respect, integrity, and accountability to create a culture of inclusion where everyone can thrive at work and beyond.
Qualifications
Required Qualifications:
- Doctorate in relevant field
- OR equivalent experience.
- Solid understanding of GPU architecture, memory hierarchies, parallel computing and algorithm optimization.
- Hands-on experience in GPU programming, including performance profiling and optimization tools.
- Advanced C++ programming skills.
Other Requirements:
Ability to meet Microsoft, customer and/or government security screening requirements are required for this role. These requirements include but are not limited to the following specialized security screenings:
- Microsoft Cloud Background Check: This position will be required to pass the Microsoft Cloud background check upon hire/transfer and every two years thereafter.
Preferred Qualifications:
- 5+ years of experience in GPU programming and optimization, expert knowledge of CUDA, ROCm, Triton, PTX, CUTLASS, or similar GPU programming frameworks
- Experience with machine learning frameworks (PyTorch, TensorFlow)
- Familiarity with compiler optimization techniques and background in auto-tuning and automated code generation
- Publication record in relevant conferences or journals (MLSys, NeurIPS, ICML, ICLR, AISTATS, ACL, EMNLP, NAACL, ISCA, MICRO, ASPLOS, HPCA, SOSP, OSDI, NSDI, etc.)
#M365Core #M365Research #Research
Responsibilities
- Design, implement, and optimize GPU kernels for complex computational workloads such as AI inferencing.
- Research and develop novel optimization techniques for generation of GPU kernels.
- Profile and analyze kernel performance using advanced diagnostic tools.
- Generate automated solutions for kernel optimization and tuning.
- Collaborate with other researchers to improve model performance.
- Document optimization strategies and maintain performance benchmarks.
- Contribute to the development of internal GPU computing frameworks.