Senior Software Engineer-AI Infra

Microsoft • Redmond, WA

5d

About The Position

The Artificial Intelligence Performance team at Microsoft develops AI software that enables running AI models everywhere, from world’s fastest AI supercomputers, to servers, desktops, mobile phones, IoT devices and internet browsers. They collaborate with hardware teams and partners, operating at the intersection of AI algorithmic innovation, purpose-built AI hardware, systems, and software. The team prides itself on a collaborative and inclusive culture. They own inference performance of OpenAI and other state-of-the-art LLM models, working directly with OpenAI on models hosted on the Azure OpenAI service, serving massive workloads with trillions of inferences per day in major Microsoft products. As a Senior Software Engineer - Performance, you will work on multiple levels of the AI software stack, including fundamental abstractions, programming models, compilers, runtimes, libraries, and APIs to enable large-scale training and inferencing of models. Responsibilities include benchmarking OpenAI and other LLM models for performance on GPUs and Microsoft HW, debugging and optimizing performance, monitoring performance, and enabling rapid deployment of these models with minimal hardware, contributing to Microsoft Azure's capex goals. This is a hands-on technical role requiring software design and development skills, seeking someone with a demonstrated history of solving technical problems and motivation to tackle challenges in building a full end-to-end AI stack. An entrepreneurial approach, initiative, and speed are essential. Microsoft's mission is to empower every person and every organization to achieve more, fostering a culture of inclusion based on values of respect, integrity, and accountability.

Requirements

• Bachelor’s degree in computer science or related technical field AND 4+ years technical engineering experience with coding in languages including, but not limited to, C, C++, C#, Java, JavaScript, or Python OR equivalent experience.

Nice To Haves

• Master's Degree or Ph.D in Computer Science or related technical field AND technical engineering experience with coding in languages including, but not limited to, C, C++, C#, Java, JavaScript, or Python OR Bachelor's Degree in Computer Science or related technical field AND 12+ years technical engineering experience with coding in languages including, but not limited to, C, C++, C#, Java, JavaScript, or Python OR equivalent experience.
• Technical background and solid foundation in software engineering principles, computer architecture, GPU architecture, HW neural net acceleration
• Experience in end-to-end performance analysis and optimization of state-of-the-art LLMs, including proficiency using GPU profiling tools
• Experience in DNN/LLM inference and experience in one or more DL frameworks such as PyTorch, Tensorflow, or ONNX Runtime and familiarity with CUDA, ROCm, Triton
• Cross-team collaboration skills and the desire to collaborate in a team of researchers and developers

Responsibilities

• Identify and drive improvements to end-to-end inference performance of OpenAI and other state-of-the-art LLMs
• Measure, benchmark performance on Nvidia/AMD GPUs and first party Microsoft silicon
• Optimize and monitor performance of LLMs and build SW tooling to enable insights into performance opportunities ranging from the model level to the systems and silicon level to improve customer experience and reduce the footprint of the computing fleet
• Enable fast time to market of LLMs/models and their deployments at scale by building SW tools that afford velocity in porting models on new Nvidia and AMD GPUs
• Design, implement, and test functions or components for our AI/DNN/LLM frameworks and tools
• Speeding up/reducing complexity of key components/pipelines to improve performance and/or efficiency of our systems
• Communicate and collaborate with our partners both internal and external
• Embody Microsoft's Culture and Values

Benefits

• Certain roles may be eligible for benefits and other compensation. Find additional benefits and pay information here: https://careers.microsoft.com/us/en/us-corporate-pay

What This Job Offers

• Job Type: Full-time
• Career Level: Senior