Principal Software Engineer

The Artificial Intelligence (AI) Frameworks team at Microsoft develops the AI software used to train and deploy the world’s most advanced AI models. We collaborate with our hardware teams and partners to build the software stacks for Microsoft’s next-generation supercomputers and the Maia AI accelerators. We work closely with ML researchers and developers to optimize and scale out model training and inference. We work with OpenAI on the models hosted on the Azure OpenAI service. As a member of this team, you will have the opportunity to work on developing and evaluating core algorithmic and hardware technologies to enable large scale inferencing and training of the most advanced AI models on novel AI hardware.

Collaborate broadly with ML researchers, system engineers, and production engineers. Engage with key partners to understand and evaluate performance and quality for state-of-the-art LLMs at different scales. Build software tools to support validation and exploration of LLM optimization technologies. Perform software development in model scripting and/or kernel languages, such as Python, C/C++, CUDA. Identify requirements, scope solutions, estimate work, schedule deliverables.