Microsoft Logo

Microsoft

Senior Software Engineer- AI Frameworks

Posted 6 Days Ago
Remote
Hiring Remotely in United States
120K-261K Annually
Senior level
Remote
Hiring Remotely in United States
120K-261K Annually
Senior level
Designs and operates production-quality AI framework, runtime, benchmarking, performance, automation, and observability components. Benchmarks, profiles, debugs, and optimizes large language model training and inference across GPUs and Microsoft hardware. The role drives projects from definition through deployment, investigates cross-stack reliability issues, partners with research, infrastructure, model, and hardware teams, contributes to technical standards, and mentors engineers.
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 Senior 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 Senior Software Engineer, you will own significant components and projects across AI performance, benchmarking, automation, and developer tooling. You will independently translate model and platform needs into robust software, investigate complex performance and reliability issues, and collaborate with partner teams to deliver measurable improvements into production. 


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
  • Design, implement, test, and operate production-quality components across AI frameworks, runtimes, benchmarking systems, performance tooling, and service integrations.  

  • Benchmark, profile, debug, and optimize large language model training and inference workloads across GPUs and Microsoft hardware.  

  • Build automation and observability that detect regressions, improve reproducibility, surface actionable insights, and accelerate model and hardware onboarding.  

  • Drive scoped projects from problem definition through deployment, balancing delivery speed, maintainability, reliability, and measurable customer or capacity impact.  

  • Partner with researchers, model teams, infrastructure owners, and hardware vendors to diagnose cross-stack issues and deliver production-ready solutions.  

  • Contribute to technical design reviews, engineering standards, operational health, and mentoring of other engineers.  

  • Embody Microsoft’s culture and values.  


Qualifications

Required/Minimum Qualifications: 

  • Bachelor’s Degree in Computer Science or a related technical field and 4+ 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: 

  • Expertise in GPU or equivalent accelerator programming and kernel optimization using CUDA, Triton, or comparable technologies. 

  • Experience using profiling and benchmarking to deliver measurable performance improvements, with an understanding of memory hierarchy, parallel execution, synchronization, and hardware utilization. 

  • Familiarity with vLLM, SGLang, or equivalent inference frameworks, including their model execution, scheduling, batching, and KV-cache mechanisms. 

  • Demonstrated technical ownership and cross-team collaboration, including independently delivering complex engineering changes into production. 

  • Ability to use AI-assisted development tools effectively while maintaining sound engineering judgment and accountability for correctness, performance, and maintainability 


#AIInfra 


Software Engineering IC4 - The typical base pay range for this role across the U.S. is USD $119,800 - $234,700 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 $160,200 - $261,000 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

12 Hours Ago
In-Office or Remote
CA, USA
152K-288K Annually
Senior level
152K-288K Annually
Senior level
Artificial Intelligence • Computer Vision • Hardware • Robotics • Metaverse
Develop production-grade integrations for NVIDIA Grove across AI frameworks including Dynamo, llm-d, Ray, and PyTorch. Build adapters, plugins, operators, runtime components, reference workflows, and developer tooling. Optimize distributed multi-node and multi-GPU training and inference, improve Kubernetes observability, define APIs, and troubleshoot containers, networking, scheduling, CUDA, and runtime issues. Collaborate with engineering teams and open-source communities to upstream changes and maintain compatibility across evolving frameworks.
Top Skills: C++Ci/CdCudaDockerDynamoGoGpuHelmKubernetesKustomizeLlm-DNsightNvidia GrovePythonPyTorchRayRpcTriton Inference Server
5 Days Ago
In-Office or Remote
United States
143K-331K Annually
Senior level
143K-331K Annually
Senior level
Software • Quantum Computing • Metaverse • Infrastructure as a Service (IaaS)
Build and operate production-quality AI framework, runtime, benchmarking, performance, automation, observability, and developer-tooling systems. Optimize large language model training and inference across GPUs and Microsoft hardware, diagnose cross-stack reliability and performance issues, and improve model onboarding, hardware utilization, Azure efficiency, and deployment speed. Senior engineers own major components and projects; Principal engineers define technical direction, architecture, and multi-release strategy while leading cross-team initiatives and mentoring technical leaders.
Top Skills: Amd GpusAzureC++CudaNvidia GpusOnnx RuntimePythonPyTorchRocmTensorFlowTriton
2 Days Ago
Remote
United States
120K-304K Annually
Senior level
120K-304K Annually
Senior level
Software • Quantum Computing • Metaverse • Infrastructure as a Service (IaaS)
Develop and operate production AI framework components, runtimes, benchmarking systems, performance tools, and integrations. Benchmark and optimize large language model training and inference across GPUs and Microsoft hardware, build automation and observability, investigate cross-stack performance and reliability issues, and deliver scalable platform improvements. Principal engineers additionally set technical direction, lead multi-team initiatives, define architectures, and mentor technical leaders.
Top Skills: Ai FrameworksAi RuntimesAzureC++CudaGpusOnnx RuntimePythonPyTorchRocmTensorFlowTriton

What you need to know about the NYC Tech Scene

As the undisputed financial capital of the world, New York City is an epicenter of startup funding activity. The city has a thriving fintech scene and is a major player in verticals ranging from AI to biotech, cybersecurity and digital media. It also has universities like NYU, Columbia and Cornell Tech attracting students and researchers from across the globe, providing the ecosystem with a constant influx of world-class talent. And its East Coast location and three international airports make it a perfect spot for European companies establishing a foothold in the United States.

Key Facts About NYC Tech

  • Number of Tech Workers: 549,200; 6% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Capgemini, Bloomberg, IBM, Spotify
  • Key Industries: Artificial intelligence, Fintech
  • Funding Landscape: $25.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Greycroft, Thrive Capital, Union Square Ventures, FirstMark Capital, Tiger Global Management, Tribeca Venture Partners, Insight Partners, Two Sigma Ventures
  • Research Centers and Universities: Columbia University, New York University, Fordham University, CUNY, AI Now Institute, Flatiron Institute, C.N. Yang Institute for Theoretical Physics, NASA Space Radiation Laboratory

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account