Sr. Software Engineer- AI/ML, AWS Neuron Distributed Training

Job Description

Senior machine learning engineer role in the AWS Neuron team responsible for developing and tuning ML models for distributed training on AWS Trainium and Inferentia silicon.

Requirements

• Bachelor's degree in computer science or equivalent
• 5+ years of professional software development experience
• Experience in machine learning, data mining, information retrieval, statistics or natural language processing

Responsibilities

• Lead efforts in building distributed training support into Pytorch, Tensorflow using XLA and Neuron compiler
• Tune ML models for highest performance on AWS Trainium and Inferentia silicon
• Work with chip architects, compiler engineers, and runtime engineers

Benefits

• Medical, dental, and vision coverage (multiple plan options)
• Health savings and flexible spending accounts
• 24/7 Employee Assistance Program for mental health support
• 401(k) with company match and various investment choices
• Company-paid life/AD&D and disability insurance
• Restricted stock units (RSUs) for employee ownership
• Child care and elder care referrals, adoption assistance
• Paid maternity and parental leave (including Ramp Back)
• Paid time off (PTO) and company holidays
• Employee discount on Amazon.com products
• Career Choice program (95% tuition and textbook coverage)