Senior ML Engineer: Distributed Training on Neuron

Amazon; Seattle (WA); On-site; USD 151,300 - 261,500; Full time

Benefits offered by this job

Comprehensive medical benefits

Flexible work-life balance

Mentorship opportunities

Job summary

A leading cloud technology company in Seattle seeks a Senior Machine Learning Engineer for the Distributed Training team. This role involves developing and optimizing ML models using AWS hardware. Candidates should have extensive experience in Python and machine learning frameworks like PyTorch and JAX. A Bachelor's degree in Computer Science is required, along with significant software development experience. The position offers a competitive salary and a commitment to work-life balance.

Qualifications

• 5+ years of non-internship professional software development experience.
• 5+ years of programming experience with at least one software language.
• Experience as a mentor or tech lead.

Responsibilities

• Lead efforts to build distributed training support into PyTorch and JAX.
• Optimize models for peak performance on AWS custom silicon.
• Work with cross-functional teams including chip architects and compiler engineers.

Skills

Python

Machine Learning

Software Development

Distributed Training

Mentoring

Education

Bachelor's degree in computer science or equivalent

Master's degree in computer science or equivalent

Tools

PyTorch

JAX

TensorFlow

Distributed Libraries

Machine Learning Frameworks

Job description

A leading cloud technology company in Seattle seeks a Senior Machine Learning Engineer for the Distributed Training team. This role involves developing and optimizing ML models using AWS hardware. Candidates should have extensive experience in Python and machine learning frameworks like PyTorch and JAX. A Bachelor's degree in Computer Science is required, along with significant software development experience. The position offers a competitive salary and a commitment to work-life balance.