Back to Jobs
AI & Machine Learning Just now

Engineering Manager, ML Training & Serving

United StatesUnited States
Full-time
$204,000 - $290,000
Senior-Level

Job Description

Key Skills Required

Master these to land this role

DevOpsBestseller 🔥
Learn in 63 Hours
Machine LearningBestseller 🔥
Learn in 42 Hours
Python ScriptingAI EngineerTensorFlow

Want to know if you're a match for this job?

Calculate My Match Score

Affirm is reinventing credit to make it more honest and friendly, giving consumers the flexibility to buy now and pay later without any hidden fees or compounding interest.

We are seeking an Engineering Manager to lead our ML Training & Serving team. You will directly lead a team of platform engineers and combine effective people leadership with strong technical judgment in ML infrastructure. Working closely with senior ICs and partners across Machine Learning and Infrastructure, you will help shape the team’s roadmap, drive execution, and ensure the platform reliably supports Affirm’s ML priorities.

What You'll Do

  • Partner with senior ICs and engineering leadership to define and execute the roadmap for ML training & serving, spanning model training, deployment workflows, GPU infrastructure, and low-latency model serving.
  • Lead, coach, and grow a team of platform engineers while staying closely engaged in technical decisions and execution.
  • Drive delivery and operational health for the team’s infrastructure, balancing reliability, developer experience, performance, and cost.
  • Evaluate and adopt modern ML infrastructure capabilities as Affirm’s needs evolve, including transformer-based workloads and GPU compute.
  • Collaborate with ML modeling, product, and infrastructure teams to ensure the platform supports Affirm’s highest-priority ML initiatives.
  • Recruit, develop, and retain high-performing platform engineers.

What We Look For

  • 7+ years of industry experience in software and/or machine learning engineering, including 2+ years managing engineers and meaningful hands-on software engineering experience.
  • Strong experience building and operating production ML or distributed systems infrastructure, with hands-on experience in at least one of model training, model serving, deployment workflows, or GPU infrastructure.
  • Solid understanding of ML data needs, including training datasets, data quality, reproducibility, and evaluation data.
  • Familiarity with modern ML workloads, such as deep learning, transformer architectures, and large-scale training or serving.
  • Strong systems thinking and the ability to partner with senior ICs on architectural trade-offs.
  • Track record of delivering platforms or infrastructure that improve the productivity and impact of engineering or ML teams.
  • Experience recruiting, coaching, and developing engineers across career stages.
  • Effective cross-functional collaboration and the ability to lead through ambiguity.
  • Experience with applied ML modeling is a plus.
  • Bachelor’s degree in a technical field or equivalent practical experience.

Employees new to Affirm typically come in at the start of the pay range. Affirm focuses on providing a simple and transparent pay structure which is based on a variety of factors, including location, experience and job-related skills.

Base pay is part of a total compensation package that may include equity rewards, monthly stipends for health, wellness and tech spending, and benefits (including 100% subsidized medical coverage, dental and vision for you and your dependents.)

USA base pay range (CA, WA, NY, NJ, CT) per year: $230,000 - $290,000
USA base pay range (all other U.S. states) per year: $204,000 - $264,000

How would you rate this job post?

See what other professionals think about this role.

Safety First

  • Never pay for a job application.
  • Do not share sensitive bank info.
  • Verify the client before starting work.
Learn More