Microsoft Logo

Microsoft

Principal AI Accelerator Tools Development Engineer

Reposted 3 Days Ago
Be an Early Applicant
In-Office
Redmond, WA, USA
143K-304K Annually
Senior level
In-Office
Redmond, WA, USA
143K-304K Annually
Senior level
Lead development of scalable stress, validation, and performance tooling for MAIA AI accelerator platforms. Build workload generation, kernel optimization, compiler/SDK integration, automated orchestration and telemetry, platform validation suites, benchmarking, and CI-driven regression infrastructure to enable platform bring-up, reliability, and performance optimization across hardware and software stacks.
The summary above was generated by AI
Overview

Microsoft Silicon, Cloud Hardware, and Infrastructure Engineering (SCHIE) is the team behind Microsoft’s expanding Cloud Infrastructure and responsible for powering Microsoft’s “Intelligent Cloud” mission. SCHIE delivers the core infrastructure and foundational technologies for Microsoft's over 200 online businesses including Bing, MSN, Office 365, Xbox Live, Teams, OneDrive, and the Microsoft Azure platform globally with our server and data center infrastructure, security and compliance, operations, globalization, and manageability solutions. Our focus is on smart growth, high efficiency, and delivering a trusted experience to customers and partners worldwide and we are looking for passionate engineers to help achieve that mission. 

Microsoft's Hardware Systems organization is developing AI-native silicon and hyperscale systems designed to power the next generation of frontier AI models. The MAIA platform combines custom silicon, high-performance networking, advanced compiler technologies, and large-scale system infrastructure to enable industry-leading AI training and inference. 

The Platform Systems Engineering (PSE) team is seeking a Principal AI Accelerator Tools Development Engineer to lead the development of next-generation stress, validation, and performance tooling for MAIA AI accelerator platforms. 

In this role, you will build software frameworks, stress workloads, and validation tools that exercise every layer of the AI stack, from hardware execution engines and memory subsystems to compiler-generated kernels, distributed communication fabrics, and large-scale AI workloads. Your work will play a critical role in platform bring-up, qualification, performance characterization, reliability validation, and fleet readiness for both current and future generations of MAIA systems. 

You will work closely with silicon architects, compiler teams, runtime developers, performance engineers, validation teams, and AI framework developers to translate platform requirements into scalable tooling and workload solutions. 

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.



Responsibilities

AI Workload & Stress Tool Development 

  • Design and develop scalable stress, performance, and validation frameworks for MAIA AI accelerator platforms. 

  • Build workload generation infrastructure capable of exercising compute, memory, interconnect, networking, storage, and system-level resources. 

  • Develop reusable stress tools using PyTorch, Triton, Python, C++, and custom MAIA SDKs. 

  • Create synthetic and production-inspired workloads that model training and inference behaviors observed in large-scale AI deployments. 

  • Build automated infrastructure for workload deployment, orchestration, telemetry collection, and result analysis. 

Hardware-Aware Workload Optimization 

  • Develop and optimize kernels targeting custom AI accelerators. 

  • Create GEMM, attention, collective communication, and memory intensive stress workloads. 

  • Analyze execution behavior across the hardware-software stack and identify bottlenecks impacting utilization and performance. 

  • Collaborate with compiler and runtime teams to improve workload efficiency and hardware utilization. 

Compiler & SDK Integration 

  • Develop tooling that integrates with MAIA compiler pipelines, SDKs, runtime environments, and performance analysis tools. 

  • Understand and debug compiler output, generated kernels, scheduling decisions, and execution behavior. 

  • Build automation around model compilation, kernel validation, regression testing, and workload portability. 

  • Partner with compiler teams to validate new compiler features and workload optimization strategies. 

Platform Validation & Reliability 

  • Design workload suites for platform bring-up, qualification, and reliability testing. 

  • Build comprehensive regression infrastructure supporting silicon, firmware, system software, and platform releases. 

  • Develop automated validation tools capable of identifying correctness, performance, thermal, power, and stability issues. 

  • Enable platform readiness through scalable validation methodologies and continuous regression testing. 

Performance Engineering 

  • Characterize system performance across compute, networking, memory, and storage subsystems. 

  • Develop benchmarking methodologies and performance dashboards. 

  • Adapt and optimize industry-standard workloads including:  

  • HPL/HPC benchmarks 

  • LLM training workloads 

  • Transformer-based inference workloads 

  • Collective communication benchmarks 

  • AI framework benchmark suites 

  • Drive root-cause analysis and optimization initiatives across the stack. 

Developer Productivity & Automation 

  • Improve developer productivity through automation, CI/CD integration, diagnostics, and debugging infrastructure. 

  • Build reusable tooling for workload generation, failure triage, telemetry analysis, and reporting. 

  • Develop dashboards and automated workflows for large-scale validation environments. 

  • Partner with engineering teams to convert recurring validation challenges into durable tooling solutions. 


Qualifications

Required Qualifications:

  • Master's Degree in Electrical Engineering, Computer Engineering, Mechanical Engineering, or related field AND 7+ years technical engineering experience
    • OR Bachelor's Degree in Electrical Engineering, Computer Engineering, Mechanical Engineering, or related field AND 8+ years technical engineering experience
    • OR equivalent experience
  • 8+ years of experience developing and optimizing AI training and inference workloads for GPUs, AI accelerators, or HPC platforms, including distributed AI systems, compute-intensive kernel development, and performance-focused software development using frameworks such as  C++, PyTorch and Triton.
  • 8+ years of experience analyzing and optimizing workloads on AI accelerator, GPU, or HPC platforms, including performance profiling, bottleneck analysis, and workload optimization, with knowledge of accelerator architectures, memory hierarchies, interconnects, runtime systems, and distributed AI infrastructure.
  • 8+ years of experience developing and optimizing GPU or AI accelerator kernels; building automated stress, validation, benchmarking, and reliability frameworks; and driving performance analysis and root-cause resolution across hardware, software, and distributed system environments.

Other Qualifications:

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. 

Preferred Qualifications:

  • Experience with AI compiler technologies and kernel generation frameworks, including LLVM, MLIR, Triton Compiler, or similar compiler toolchains.
  • Experience training, optimizing, or deploying large-scale AI models, including LLM training and inference workloads.
  • Experience with custom AI accelerator SDKs, collective communication libraries, and large-scale distributed computing environments.
  • Experience supporting silicon bring-up, platform qualification, post-silicon validation, or hardware/software integration activities and cloud-scale validation infrastructure

#azure #MAIA #AI/ML


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.

HQ

Microsoft Redmond, Washington, USA Office

1 Microsoft Way, Redmond, WA, United States, 98052

Microsoft Bellevue, Washington, USA Office

Bellevue, United States

Microsoft Issaquah, Washington, USA Office

Issaquah, United States

Microsoft Kirkland, Washington, USA Office

Kirkland, United States

Microsoft Mercer Island, Washington, USA Office

Mercer Island, United States

Microsoft Sammamish, Washington, USA Office

Sammamish, United States

Microsoft Seattle, Washington, USA Office

Seattle, United States

Similar Jobs

17 Minutes Ago
Easy Apply
Remote or Hybrid
USA
Easy Apply
151K-215K Annually
Senior level
151K-215K Annually
Senior level
Cloud • Information Technology • Security • Software • Cybersecurity
Lead digital transformation and AI-powered support initiatives to improve case lifecycle, self-service adoption, and customer outcomes. Own knowledge management (KCS), process optimization (Lean/Six Sigma), enablement, COE programs, and cross-functional partnerships with Engineering, Product, and Operations. Analyze case trends and metrics to drive product, process, and digital support improvements and reduce resolution times.
Top Skills: Ai-Enabled Support ToolsAi/MlAsanaConfluenceJIRAKcsSFDCTableau
An Hour Ago
Remote or Hybrid
United States
81K-122K Annually
Senior level
81K-122K Annually
Senior level
Artificial Intelligence • Automotive • Greentech • Information Technology • Machine Learning • Software • Cybersecurity
Partner with Retail Consumer Engineering and Product leaders to deliver HR services: organizational design, talent management, workforce planning, change management, employee engagement, compliance, and people analytics. Advise leaders, support recruiting and retention, lead performance and compensation processes, and resolve employee relations issues while ensuring policy and legal compliance.
Top Skills: Excel
An Hour Ago
Remote or Hybrid
United States
89K-134K Annually
Senior level
89K-134K Annually
Senior level
Artificial Intelligence • Automotive • Greentech • Information Technology • Machine Learning • Software • Cybersecurity
Lead planning, sequencing, and delivery of Employee Service Center and People Solutions Operations initiatives. Manage portfolio roadmap and intake, run end-to-end projects, govern risks/dependencies, enable BAU transitions, drive stakeholder alignment and change communications, and support continuous improvement of PM practices.
Top Skills: Ai-Enabled ToolsExcelMicrosoft PowerpointMicrosoft TeamsMicrosoft WordSmartsheetWorkday

What you need to know about the Seattle Tech Scene

Home to tech titans like Microsoft and Amazon, Seattle punches far above its weight in innovation. But its surrounding mountains, sprinkled with world-famous hiking trails and climbing routes, make the city a destination for outdoorsy types as well. Established as a logging town before shifting to shipbuilding and logistics, the Emerald City is now known for its contributions to aerospace, software, biotech and cloud computing. And its status as a thriving tech ecosystem is attracting out-of-town companies looking to establish new tech and engineering hubs.

Key Facts About Seattle Tech

  • Number of Tech Workers: 287,000; 13% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Amazon, Microsoft, Meta, Google
  • Key Industries: Artificial intelligence, cloud computing, software, biotechnology, game development
  • Funding Landscape: $3.1 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Madrona, Fuse, Tola, Maveron
  • Research Centers and Universities: University of Washington, Seattle University, Seattle Pacific University, Allen Institute for Brain Science, Bill & Melinda Gates Foundation, Seattle Children’s Research Institute

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account