Apply on Employer Site

ChatGPT Jobs · 5 days ago

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

Seattle, WA

Full-time

Hybrid

Senior Level

$139K/yr - $184K/yr

5+ years exp

Annapurna Labs is a company that designs silicon and software to accelerate innovation, and they are seeking a Senior Machine Learning Engineer for their Distributed Training team. The role involves developing, enabling, and performance tuning machine learning model families, particularly focusing on large-scale models and distributed training solutions using AWS custom silicon.

Computer Software

Responsibilities

You will lead efforts to build distributed training support into PyTorch, the Neuron compiler, and runtime stacks

You will enable distribute training strategies as well as use them to optimize models to achieve peak performance and maximize efficiency on AWS custom silicon, including Trainium servers

Qualification

Machine LearningPyTorchDistributed TrainingSoftware DevelopmentComputer ArchitectureDesign PatternsCoding StandardsSource Control ManagementMentorshipTeam Leadership

Required

Bachelor's degree in computer science or equivalent

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 (design patterns, reliability and scaling) of new and existing systems experience

5+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience

Experience as a mentor, tech lead or leading an engineering team

Experience in machine learning, large scale training with LLMs and expertise in Pytorch