Senior AI Software Architect

Microsoft Microsoft · Big Tech · Redmond, WA +4 · Software Engineering

Senior AI Software Architect role focused on optimizing AI accelerator platforms, including kernels, compiler, runtime, distributed training, and inference serving. The role involves analyzing workload behavior, identifying bottlenecks, and guiding hardware-software co-design for large-scale AI deployment.

What you'd actually do

  1. Lead end-to-end software architecture and performance optimization for AI accelerator platforms.
  2. Prototype and validate software capabilities across kernels, compiler and runtime layers, distributed training and inference frameworks, and serving infrastructure.
  3. Analyze workload behavior at scale to identify performance bottlenecks, numerical correctness issues, and system-level efficiency opportunities.
  4. Use workload insights to guide hardware-software co-design decisions across architecture, silicon, systems software, networking, and product teams.
  5. Define requirements, evaluate tradeoffs, and deliver practical solutions for production-scale AI systems.

Skills

Required

  • C
  • C++
  • C#
  • Java
  • JavaScript
  • Python
  • systems software
  • distributed systems

Nice to have

  • PhD in Computer Science, Computer Architecture, Electrical Engineering, Machine Learning, High-Performance Computing, or a related field
  • Master's Degree in Computer Science, Electrical Engineering, Computer Engineering, or related field
  • Experience designing, building, or optimizing systems software for AI, machine learning, high-performance computing, or distributed systems.
  • Experience analyzing large-scale AI training runs, including loss curves, convergence behavior, gradient flow, activation statistics, and numerical stability.
  • Experience debugging training correctness issues such as gradient divergence, NaNs/Infs, optimizer behavior, mixed-precision instability, distributed synchronization bugs, or hardware/software numerical differences.

What the JD emphasized

  • performance optimization
  • AI accelerator platforms
  • distributed training
  • inference serving
  • numerical correctness issues
  • hardware-software co-design

Other signals

  • AI accelerator platforms
  • distributed training
  • inference serving
  • hardware-software co-design