SDE I, ML Infra & AI Accelerators

Amazon Web Services (AWS); Cupertino (CA); On-site; USD 127,000 - 185,000; Full time

Job summary

Annapurna Labs (U.S.) Inc. in Cupertino, CA, is seeking a Software Development Engineer to build and evolve ML tooling for AWS Inferentia/Trainium accelerators. You’ll work across the stack with hardware engineers, architects, and ML researchers on end-to-end tooling and optimization.

You will design and implement new pipelines, automation, and tools, participate in design reviews, and contribute to a fast-paced startup-like team transforming AI infrastructure at scale.

Qualifications

• 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 language such as Java, Python, C++, C#, Go, Rust, or TypeScript.
• Experience with data structure implementation, basic algorithm development, and OO design principles.
• Proficiency in Java and at least one of Go, Python, or TypeScript.
• Familiarity with Git and CI/CD pipelines.

Responsibilities

• Lead design and implementation of new tools, pipelines and automation for ML workloads on AI accelerators.
• Collaborate with hardware engineers, system architects and researchers to ensure tool compatibility.
• Build high-impact solutions for a large and growing customer base.
• Participate in design discussions, code reviews, and cross-functional collaboration.

Skills

Java

Python

C++

Go

TypeScript

Tools

Git

CI/CD

Job description

Annapurna Labs (U.S.) Inc. in Cupertino, CA, is seeking a Software Development Engineer to build and evolve ML tooling for AWS Inferentia/Trainium accelerators. You’ll work across the stack with hardware engineers, architects, and ML researchers on end-to-end tooling and optimization.

You will design and implement new pipelines, automation, and tools, participate in design reviews, and contribute to a fast-paced startup-like team transforming AI infrastructure at scale.