Drimva

Amazon

Software Development Manager, SageMaker Training

$213K–$288K

Santa Clara, CA · Manager · Full-time · Posted Oct 10, 2026

Experience
3+ years engineering team management; 7+ years working directly within engineering teams; 5+ years designing or architecting systems.

About the job

Leads the team building and operating SageMaker services that let customers post-train and customize foundation models at scale.

What you'll do

  • Own the team's charter, roadmap, technical direction, and decisions about what to build.
  • Hire, develop, and retain engineers, partnering with senior engineers, applied scientists, and product managers.
  • Own production service health and improve training throughput and cost across accelerator fleets.

What they ask for

  • 3+ years of engineering team management experience.
  • 7+ years of working directly within engineering teams experience.
  • 5+ years of designing or architecting new and existing systems, including design patterns, reliability, and scaling.
  • Knowledge of full software/hardware/networks development life-cycle practices; experience partnering with product or program management teams and recruiting, hiring, mentoring/coaching, and managing Software Engineers.

Nice to have

  • Experience with Machine Learning and Large Language Model fundamentals, including architecture, training/inference lifecycles, and model execution optimization.
  • Experience with PyTorch, JIT compilation, and AOT tracing; deep learning libraries, production LLM deployment, or large-scale machine learning systems.

Pay

212,700.00 - 287,700.00 USD annually (base salary)

Tools

PyTorch, Verl, FSDP, Megatron, vLLM, TensorFlow, MxNet, Neuron

AI focus

Leads development of managed post-training and model-customization services, including reinforcement learning and agentic training; improves training performance and cost on accelerator fleets.

Model-written summary · the original posting is authoritative

First seen Oct 15, 2026 · Last verified Nov 4, 2026 (UTC)