Software Development Manager, AWS Neuron SDK - Distributed Training

Job overview

The Software Development Manager, AWS Neuron SDK - Distributed Training role is an onsite position based in Cupertino, CA. Compensation is not specified in the posting, and this role supports the Neuron Training team in enabling ML training at scale. This role exists to lead a team in developing integrations for distributed training libraries. The manager will drive performance optimization, model onboarding, and technical strategy for frontier model architectures on Trainium hardware.

What you'll do

• Lead engineers focused on ML training
• own customer onboarding journey
• drive initiatives to maximize MFU
• build tooling and automation
• partner with compiler and runtime teams
• develop scalable processes
• recruit and mentor team.

What we're looking for

Skills & competencies

full-time

day shift

cupertino ca

distributed training

machine learning

leadership

trainium

aws

neuron sdk

onsite

Work arrangement

• Weekend coverage required

Benefits & perks

• Health insurance, 401(k) matching, paid time off, parental leave, sign-on payments, and restricted stock units (RSUs).

Why this role

Focuses on distributed training and ML performance optimization.

About the employer

Amazon Development Center U.S., Inc. is hiring for this role. Industry: Software Publishers. Sector: 51.