New 
Principal Software Engineer - AI Frameworks
 Microsoft | |
 United States, California, Mountain View | |
 Aug 14, 2025 | |
|
OverviewThe Artificial Intelligence (AI) Frameworks team at Microsoft develops AI software that enables running AI models everywhere, from world's fastest AI supercomputers, to servers, desktops, mobile phones, internet of things (IoT) devices and internet browsers. We collaborate with our hardware teams and partners, both internal and external, and operate at the intersection of AI algorithmic innovation, purpose-built AI hardware, systems, and software. We are a team of highly capable and motivated people that pride themselves on a collaborative and inclusive culture. We own inference performance of OpenAI and other state of the art large language model (LLM) models and work directly with OpenAI on the models hosted on the Azure OpenAI service serving some of the largest workloads on the planet with trillions of inferences per day in major Microsoft products, including Office, Windows, Bing, SQL Server, and Dynamics. As a Principal Software Engineer on the team, you will have the opportunity to work on multiple levels of the AI software stack, including the fundamental abstractions, programming models, compilers, runtimes, libraries and application programming interfaces (APIs) to enable large scale training and inferencing of models. You will benchmark OpenAI and other LLM models for performance on graphics processing units (GPUs) and Microsoft hardware, debug and optimize performance, monitor performance and enable these models to be deployed in the shortest amount of time and the least amount of hardware possible helping achieve Microsoft Azure's capex goals. This is a hands-on technical role requiring software design and development skills. We're looking for someone who has a demonstrated history of solving 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. 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.
ResponsibilitiesIdentify and drive improvements to end-to-end inference performance of OpenAI and other state of the art LLMsMeasure, benchmark performance on Nvidia/AMD GPU's and first party Microsoft siliconOptimize and monitor performance of LLMs and build software tooling to enable insights into performance opportunities ranging from the model level to the systems and silicon level, help reduce the footprint of the computing fleet and achieve Azure AI capex goalsEnable fast time to market of LLMs/models and their deployments at scale by building software tools that afford velocity in porting models on new Nvidia, AMD GPUs and Maia siliconDesign, implement, and test functions or components for our AI/deep neural networks (DNN)/LLM frameworks and toolsSpeeding up/reducing complexity of key components/pipelines to improve performance and/or efficiency of our systemsCommunicate and collaborate with our partners both internal and external Embody Microsoft's Culture and Values | |