Senior Software Engineer- AI Frameworks

Microsoft
San Francisco, CA, United States
8 days ago
Apply on diversityjobs.com
Prepare application

Role details

Contract type
Permanent contract
Employment type
Full-time (> 32 hours)
Experience level
Expert
Experience required
4 years minimum
Compensation
$119,800.0 - $234,700.0
Working hours
Regular working hours

Tech stack

Artificial Intelligence Build Automation Microsoft Online Services C++ (Programming Language) Compilers Nvidia CUDA Computer Programming Software Debugging Programming Tools Python (Programming Language) Performance Tuning Software Tools
+3 more
Software Engineering Software Technical Review Graphics Processing Unit (GPU)

Job description

The Microsoft AI Frameworks team, developsthe software, performance systems, and engineering tools that enablestate-of-the-artAI 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 Engineerspartner 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 aSenior 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 shape what’s next for everyone.

Responsibilities

  • Design, implement, test, andoperateproduction-quality components across AI frameworks, runtimes, benchmarking systems, performance tooling, and service integrations.

  • Benchmark, profile, debug, andoptimizelarge 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.

Requirements

  • Bachelor’s Degree in Computer Scienceor 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.

Apply for this position

This job is hosted externally. Click below to view the full posting and apply.

Apply on diversityjobs.com
Prepare application

Good distractions

Talks and stories from around this role — technically off-topic, practically not.

2:14 min

Exploring internal AI product initiatives and global engineering roles

Maria Apazoglou · Coffee With Developers

1:51 min

Evolution of custom compilers and virtual machines

Florian Rappl · LIVE

2:50 min

Electronic diagnostic software tools and factory production flashing

Denis Grahovac · World Congress 2021

6:21 min

Previewing upcoming hardware acceleration capabilities for Python environments

Chris Heilmann Chris Heilmann +2 · LIVE

2:18 min

Redefining developer roles and software architecture requirements

Laurie Voss Laurie Voss · Coffee With Developers

3:00 min

Accelerating machine learning research with optimized compilers

Tanmay Bakshi · LIVE

Videos

See all

Related articles

See all