Sr. Software Engineer- AI/ML, AWS Neuron Distributed Training
Compensation
$193k – $261k /yr
Employment type
Full-time
Work setting
On-site
Location
Cupertino, CA
Schedule
Day shift
Posted
You'll be redirected to the employer's application page.
Job overview
Compensation is $193,000–$261,000 per year. The Sr. Software Engineer- AI/ML, AWS Neuron Distributed Training role is an onsite position based in Cupertino, CA. 00 - 261,500.00 USD annually, and this role supports the Annapurna Labs team within AWS, focusing on distributed training. This role exists to develop and optimize large-scale ML model training. It contributes to the team's goal of delivering cost-effective, performant machine learning solutions on Trainium-based systems.
What you'll do
- Design and optimize distributed training solutions
- extend and optimize frameworks like FSDP and torchtitan
- develop mixed-precision training techniques
- profile and tune training pipelines
- collaborate with hardware and compiler teams.
What we're looking for
- Skills & competencies
- full-timeday shiftcupertino caannapurna labssenior software engineerdistributed trainingpytorchllmoptimizationml$193k-$261k
- 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
Focus on distributed training, mixed-precision optimization, and large-scale LLM training.
About the employer
Annapurna Labs (U.S.) Inc. is hiring for this role. Industry: Custom Computer Programming Services. Sector: 54.
Additional details
- Industry sector
- 54
- Industry
- Custom Computer Programming Services
- Occupation code
- 15-1252.00
You'll be redirected to the employer's application page.
Listing ID: a2584735-b3bd-4fee-a2a0-4e589233885d