Microsoft's AI Frameworks team is looking to develop state-of-the-art AI compilers and design new AI programming abstractions for Microsoft's next-generation supercomputers and the Maia-200 AI accelerator.
Requirements
- Coding in languages including, but not limited to, C, C++, or Python
- Experience building compilers, using compiler frameworks like LLVM/MLIR
- Experience optimizing AI/numerical workloads
- Experience in numerical high-performance computing
- Experience with AI frameworks, AI inference systems, AI kernel development, and AI workload optimization
- 3+ years experience programming in C/C++
- 1+ years experience with optimized systems‑level programming
Responsibilities
- Invent and implement innovative compiler features and advanced optimization passes, leveraging tools such as LLVM, MLIR, Torch Dynamo, and Triton.
- Develop code generation techniques for new hardware platforms.
- Design and develop cutting edge AI software in C++ and Python.
- Optimize AI workloads.
- Design new programming abstractions for AI.
- Collaborate broadly across multiple disciplines from hardware architects to ML developers.
- Identify requirements, plan and design solutions, estimate effort, and schedule deliverables.
Other
- Bachelor's Degree in Computer Science or related technical field AND 2+ years technical engineering experience
- Ability to meet Microsoft, customer and/or government security screening requirements
- Microsoft Cloud Background Check: This position will be required to pass the Microsoft Cloud background check upon hire/transfer and every two years thereafter.
- Master's Degree in Computer Science or related technical field AND 6+ years technical engineering experience
- Embody Microsoft's culture and values