Senior AI/ML Software Engineer - Distributed Training

Amazon; Seattle (WA); On-site; USD 120,000 - 160,000; Full time

Benefits offered by this job

Health insurance

401(k) matching

Paid time off

Parental leave

Job summary

Amazon is seeking a Sr. Software Engineer for their AI/ML team in Seattle, WA. This position focuses on optimizing distributed training performance for AWS Neuron and requires a deep expertise in software development, machine learning, and collaborative work across various teams.

The ideal candidate will have at least 5 years of relevant experience and will be part of a supportive environment focused on mentoring and career growth. This role also includes a competitive salary and comprehensive benefits.

Qualifications

• 5+ years of non-internship professional software development experience.
• 5+ years of programming with at least one software programming language experience.
• 5+ years of leading design or architecture of new and existing systems experience.

Responsibilities

• Lead efforts to optimize distributed training performance on Trainium.
• Work across PyTorch, JAX, and the Neuron compiler to enable large-scale training.
• Collaborate with chip architects, compiler engineers, and runtime engineers.

Skills

Software Development

Programming Languages

System Design

Mentorship

Machine Learning

Education

Bachelor's degree in Computer Science or equivalent

Tools

PyTorch

JAX

Job description

Amazon is seeking a Sr. Software Engineer for their AI/ML team in Seattle, WA. This position focuses on optimizing distributed training performance for AWS Neuron and requires a deep expertise in software development, machine learning, and collaborative work across various teams.

The ideal candidate will have at least 5 years of relevant experience and will be part of a supportive environment focused on mentoring and career growth. This role also includes a competitive salary and comprehensive benefits.