Amazon.com Inc logo

Software Development Manager, SageMaker Training

Amazon.com Inc
  • Santa Clara, CA
    2 days ago

    Job Description

    Interested in building the distributed systems that let customers customize foundation models at scale? The SageMaker Training and Model Customization team builds the services customers use to fine-tune and post-train models against their own accuracy, reliability, and cost targets. We are hiring a Software Development Manager to lead the team that owns those capabilities.

    Model customization is the fastest-moving layer of the machine learning stack. Post-training has moved from supervised fine-tuning to reinforcement learning against verifiable rewards, and from single-turn tasks to agentic training where a model learns by acting in an environment over long trajectories. Each shift changes the shape of the workload. Reinforcement learning puts an inference engine inside the training loop. Agentic training adds environments and tool calls, and moves the bottleneck from one run to the next. Your team turns each new technique into a capability customers can use, without rebuilding the platform every time the research moves.

    Underneath the techniques, this stays a hard distributed systems problem. Training jobs run across large accelerator fleets for days at a time, and a single node failure can stall the job. The team works in PyTorch and across frameworks including Verl, FSDP, Megatron and vLLM, and contributes upstream to the open source projects the platform depends on.

    As the manager you own the team and its charter. You will hire and grow engineers, set technical direction with your senior engineers and product managers, and decide what the team does and does not build. You own the roadmap and defend its tradeoffs with leadership. You own the service in production, including on-call health, operational metrics, and the requests that come with sitting in the critical path of customer training workloads.

    Key job responsibilities

    • Own the team"s charter and roadmap, and defend its tradeoffs with senior leadership.
    • Hire, develop and retain engineers in a specialized field with a scarce talent pool.
    • Set technical direction with your senior engineers, applied scientists and product managers.
    • Decide what the team builds, and what it does not.
    • Turn new post-training techniques into supported customer capabilities on a predictable schedule.
    • Own the service in production, including on-call health, availability and operational metrics.
    • Improve training throughput and cost per run across large accelerator fleets.

    A day in the life

    Your morning starts in a design review. The team is working out how to keep an inference engine and a training engine in the same loop for reinforcement learning without leaving accelerators idle, and the answer will shape the future of the platform. You deep dive into possible scenarios, because at this scale a node dies routinely and recovery has to be normal rather than exceptional.

    Then a charter call. A new post-training technique is three months out of the research literature and two customers are already asking for it. With your senior engineers and product manager you decide whether it becomes a first-class capability, a recipe on top of what you already have, or a no. That decision is yours to make, and the industry does not wait for it.

    You spend an hour in 1:1s, including a career conversation with an engineer who wants to own the parallelism work. After lunch you sit a hiring debrief and make the call. Later you meet the applied scientists to understand what they have validated, and you argue about sequencing.

    You close the day writing a page for leadership: what you are not building this half, and why that is the right trade.

    About the team

    We build the managed services customers use to customize foundation models. A customer brings a task, a dataset, and a way to score a good answer. Our services run the post-training loop on their behalf. We own that entire path, from the customer-facing API down to the training session infrastructure and the accelerator fleet it runs on.

    That makes this a foundational charter. Model customization is how a customer turns a general model into one that is theirs, and every customization experience on SageMaker is built on what we deliver. When we add a technique, it becomes available to every customer on the platform. When we make the training loop faster, every customer"s job gets cheaper.

    Numbers & Facts

    LocationSanta Clara, CA
    IndustryRetail
    Company Size10,000 employees or more
    Year Founded1994
    Websitehttp://Amazon.com/militaryroles

    About Company

    At Amazon, we don’t wait for the next big idea to present itself. We envision the shape of impossible things and then we boldly make them reality. So far, this mindset has helped us achieve some incredible things. Let’s build new systems, challenge the status quo, and design the world we want to live in. We believe the work you do here will be the best work of your life.

    Wherever you are in your career exploration, Amazon likely has an opportunity for you. Our research scientists and engineers shape the future of natural language understanding with Alexa. Fulfillment center associates around the globe send customer orders from our warehouses to doorsteps. Product managers set feature requirements, strategy, and marketing messages for brand new customer experiences. And as we grow, we’ll add jobs that haven’t been invented yet.

    It’s Always Day 1
    At Amazon, it’s always “Day 1.” Now, what does this mean and why does it matter? It means that our approach remains the same as it was on Amazon’s very first day – to make smart, fast decisions, stay nimble, invent, and stay focused on delighting our customers. In our 2016 shareholder letter, Amazon CEO Jeff Bezos shared his thoughts on how to keep up a Day 1 company mindset. “Staying in Day 1 requires you to experiment patiently, accept failures, plant seeds, protect saplings, and double down when you see customer delight,” he wrote. “A customer-obsessed culture best creates the conditions where all of that can happen.” You can read the full letter here

    Our Leadership Principles
    Our Leadership Principles help us keep a Day 1 mentality. They aren’t just a pretty inspirational wall hanging. Amazonians use them, every day, whether they’re discussing ideas for new projects, deciding on the best solution for a customer’s problem, or interviewing candidates. To read through our Leadership Principles from Customer Obsession to Bias for Action, visit https://www.amazon.jobs/principles

    Skills

    • Application Programming Interface (API)unmatched
    • Customer Relationsunmatched
    • Customer Support/Serviceunmatched
    • Customer Trainingunmatched
    • Data Setsunmatched
    • Distributed Computingunmatched
    • Inference Engineunmatched
    • Leadershipunmatched
    • Machine Learningunmatched
    • Metricsunmatched
    • On Callunmatched
    • Open Sourceunmatched
    • Product Managementunmatched
    • Reinforcement Learningunmatched
    • Software Developmentunmatched
    • Team Lead/Managerunmatched
    • Vehicle Fleetsunmatched

    Be found by employers

    5,500+ employers search our resume database daily. Add yours to get found by recruiters looking for candidates like you.

    Level up your application

    Professional resume templates

    Browse dozens of recruiter approved resume templates, layouts and formats. Choose your favorite and make it your own in minutes.

    Free resume templates

    Free resume builder

    Improve your existing resume or start from scratch and create a standout, ATS-friendly resume. Add job-specific content, download and apply.

    Free resume builder