header alt image test
Microloft Logo

Microloft

Principal Software Engineer- AI Frameworks

Posted 6 Days Ago
Be an Early Applicant
Remote
Hiring Remotely in United States
143K-304K Annually
Senior level
Remote
Hiring Remotely in United States
143K-304K Annually
Senior level
Lead architecture and hands-on implementation for AI frameworks, performance systems, benchmarking, compilers, runtimes, and developer tools. Drive cross-team initiatives that improve LLM onboarding, inference performance, reliability, hardware utilization, and Azure capacity efficiency. Conduct complex systems investigations, establish scalable measurement and observability, influence technical strategy, mentor engineers, and deliver reusable production platforms across models, infrastructure, and silicon.
The summary above was generated by AI
Overview

The Microsoft AI Frameworks team, develops the software, performance systems, and engineering tools that enable state-of-the-art AI models to run reliably and efficiently at cloud scale. We work across model architectures, frameworks, compilers, runtimes, libraries, observability, benchmarking, and hardware platforms—including NVIDIA and AMD GPUs and Microsoft silicon. Our Principal Software Engineers partner with model developers, researchers, hardware teams, and production services to accelerate model onboarding, improve performance and reliability, reduce deployment time and hardware footprint, and turn performance insights into durable platform capabilities.  


This is a hands-on individual-contributor role for engineers who enjoy solving ambiguous, end-to-end systems problems. Successful candidates combine strong software engineering fundamentals with curiosity about AI workloads, disciplined measurement, and a willingness to work across organizational boundaries to deliver production impact.  


As a Principal Software Engineer, you will set technical direction and lead high-impact initiatives spanning AI frameworks, performance engineering, benchmarking, and tooling. You will identify systemic opportunities, align teams around durable architectures and success measures, and remain hands-on in the design and delivery of solutions that improve model velocity, platform efficiency, and production reliability. 


At Microsoft, our mission to empower every person and every organization on the planet to achieve more guides how we partner with customers to deliver trusted, impactful solutions. With a growth mindset culture, we innovate responsibly and measure success by shared progress people, teams, and customers. Join us to do meaningful work that changes the world and helps shapewhat’snext for everyone. 


Responsibilities
  • Define technical vision, architecture, and multi-release strategy for critical AI framework, performance, benchmarking, or developer-productivity capabilities.  

  • Lead ambiguous, cross-stack investigations and investments spanning models, frameworks, compilers, runtimes, systems, services, and silicon.  

  • Establish common measurement, automation, observability, and engineering mechanisms that turn one-off analyses into scalable platform capabilities.  

  • Drive measurable improvements in model onboarding velocity, runtime performance, reliability, hardware utilization, and Azure capacity efficiency.  

  • Influence architecture and priorities across teams; align researchers, product groups, infrastructure owners, and hardware partners around clear decisions and execution plans.  

  • Provide hands-on technical leadership through prototypes, critical-path implementation, design and code reviews, complex debugging, and operational readiness.  

  • Raise the engineering bar by mentoring engineers, developing technical leaders, and advancing standards for quality, maintainability, and inclusive collaboration.  

  • Embody Microsoft’s culture and values.  


Qualifications

Required/Minimum Qualifications: 

  • Bachelor’s Degree in Computer Science or a related technical field and 6+ years of technical engineering experience coding in languages such as C++, or Python, or equivalent experience. 


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. 


Additional or preferred qualifications: 

  • Deep expertise in GPU or equivalent accelerator programming, compilation, and low-level execution, including intermediate representations, lowering, code generation, instruction-level behavior, memory hierarchy, and synchronization. 

  • Expertise in parallelism strategies used in LLM training and inference, including tensor, pipeline, data, and expert parallelism, with the ability to evaluate their suitability for different models and hardware configurations. 

  • Expertise in distributed inference acceleration, including prefill/decode disaggregation, KV-cache transfer, collective communication, and compute/communication overlap. 

  • Strong understanding of LLM serving architectures such as vLLM, SGLang, or equivalent systems, with a track record of translating architectural improvements into measurable production gains. 

  • Demonstrated leadership of cross-team technical initiatives from strategy and design through implementation, deployment, and sustained production impact. 

  • A track record of creating reusable platforms, influencing stakeholders, and mentoring engineers. 

  • Ability to use AI-assisted development tools effectively and establish practices that improve engineering productivity without compromising correctness, performance, or maintainability. 


#AIInfra


Software Engineering IC5 - The typical base pay range for this role across the U.S. is USD $142,800 - $274,800 per year. There is a different range applicable to specific work locations, within the San Francisco Bay area and New York City metropolitan area, and the base pay range for this role in those locations is USD $188,000 - $304,200 per year.

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


This position will be open for a minimum of 5 days, with applications accepted on an ongoing basis until the position is filled.



Microsoft is an equal opportunity employer. All qualified applicants will receive consideration for employment without regard to age, ancestry, citizenship, color, family or medical care leave, gender identity or expression, genetic information, immigration status, marital status, medical condition, national origin, physical or mental disability, political affiliation, protected veteran or military status, race, ethnicity, religion, sex (including pregnancy), sexual orientation, or any other characteristic protected by applicable local laws, regulations and ordinances. If you need assistance with religious accommodations and/or a reasonable accommodation due to a disability during the application process, read more about requesting accommodations.

Similar Jobs at Microloft

6 Days Ago
Remote
United States
120K-304K Annually
Senior level
120K-304K Annually
Senior level
Automation
Develop and operate production-quality AI framework, runtime, benchmarking, performance, automation, and observability systems. Benchmark, profile, debug, and optimize large language model workloads across GPUs and Microsoft hardware. Lead cross-stack investigations, improve model onboarding, reliability, hardware utilization, and Azure capacity efficiency, while collaborating with researchers, infrastructure teams, hardware partners, and engineers. Principal-level responsibilities include setting technical direction, defining architectures, leading strategic initiatives, and mentoring technical leaders.
Top Skills: Amd GpusAzureC++CudaNvidia GpusOnnx RuntimePythonPyTorchRocmTensorFlowTriton
6 Days Ago
In-Office or Remote
3 Locations
143K-331K Annually
Senior level
143K-331K Annually
Senior level
Automation
Develop and lead production AI infrastructure across frameworks, runtimes, benchmarking, performance tooling, observability, and hardware platforms. Responsibilities include optimizing large language model training and inference, building automation, diagnosing cross-stack performance issues, defining technical architecture, driving platform improvements, collaborating across research and infrastructure teams, and mentoring engineers. The role spans senior and principal-level individual contributor responsibilities focused on scalable, reliable, and efficient AI systems.
Top Skills: Amd GpusAzureC++CudaMicrosoft SiliconNvidia GpusOnnx RuntimePythonPyTorchRocmTensorFlowTriton
One Month Ago
Remote
United States
143K-331K Annually
Senior level
143K-331K Annually
Senior level
Automation
Design, implement, and ensure quality of systems-level software, tools, and SDK integration for AI frameworks. Collaborate with hardware, ML, and platform teams; write high-performance code (C/C++, C#, Python); define architectures, coding standards, and schedules; and work on device interfaces, runtime systems, and observability/profiling.
Top Skills: AzureCC#C++DmaDriver/Kernel InterfacesEbpfGpuPciePerformance CountersPythonRdmaRustSdkTelemetry HooksTpuTrace Buffers

What you need to know about the San Francisco Tech Scene

San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.

Key Facts About San Francisco Tech

  • Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Google, Apple, Salesforce, Meta
  • Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
  • Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
  • Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account