ML Accelerator Tools SDE I

Amazon; Cupertino (CA); On-site; USD 150,000 - 190,000; Full time

Job summary

Annapurna Labs in Cupertino, CA is seeking a Software Development Engineer to help build and evolve ML tools that run and analyze workloads on AWS accelerators. You will work across the stack with hardware engineers, system architects, and ML researchers to create and optimize tooling for large-scale AI workloads.

You will lead design discussions, implement automation, and contribute to high-impact projects that accelerate the Neuron and related AI accelerator teams within AWS.

Qualifications

• Are 18 years of age or older.
• Experience with at least one modern language such as Java, Python, C++, or C# including object-oriented design.
• Experience with at least one general-purpose programming language such as Java, Python, C++, C#, Go, Rust, or TypeScript.
• Experience with data structure implementation, basic algorithm development, and/or object-oriented design principles.
• Proficiency in Java and at least one of Go, Python, or TypeScript.
• Familiarity with Git and CI/CD pipelines.

Responsibilities

• Lead the design and implementation of new tools, pipelines and automation.
• Build high-impact solutions that ship to a large and growing customer base.
• Participate in design discussions, code reviews, and cross-functional collaboration with hardware, software, and customer-facing teams.
• Create metrics, implement automation, and resolve root causes of software defects.
• Work in a startup-like environment where you're always focused on the most important problems.

Skills

Java

Python

C++

C#

Go

TypeScript

Object-oriented design

Data structures

Algorithms

Tools

Git

CI/CD pipelines

Job description

Annapurna Labs in Cupertino, CA is seeking a Software Development Engineer to help build and evolve ML tools that run and analyze workloads on AWS accelerators. You will work across the stack with hardware engineers, system architects, and ML researchers to create and optimize tooling for large-scale AI workloads.

You will lead design discussions, implement automation, and contribute to high-impact projects that accelerate the Neuron and related AI accelerator teams within AWS.