Infra SDE II: ML Accelerator Cloud Tools Lead

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

Benefits offered by this job

Health insurance

401(k) matching

Paid time off

Job summary

Annapurna Labs (U.S.) Inc. in Cupertino, CA is seeking a Software Engineer to lead the development of services that optimize, analyze, and release ML workloads and artifacts. You will drive distributed systems initiatives from architecture through delivery to customers.

The role requires deep knowledge of optimization, resource management, scheduling, and experience with EC2/EKS/Lambda or equivalent cloud services. You will work in a startup‑like environment with cross‑functional collaboration.

Qualifications

• 3+ years of non‑internship professional software development experience.
• 2+ years of non‑internship design or architecture (design patterns, reliability and scaling) of new and existing systems experience.
• Experience programming with at least one software programming language.
• Knowledge of system performance, memory management, and parallel computing principles.
• Experience in debugging, profiling, and implementing software engineering best practices in large‑scale systems, or experience debugging, profiling, and implementing best software engineering practices in large‑scale systems.
• Experience with AWS or cloud technologies.
• 3+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience.
• Knowledge of fundamentals of networking, security, databases (relational or NoSQL), operating systems (Unix, Linux, and/or Windows).
• Fundamentals of Machine learning and LLMs, their architecture along with work experience on certain LLM models.

Responsibilities

• Lead the design and implementation of new tools, pipelines and automation; work with developers, system architects, hardware engineers and users both within and external to Amazon to ensure compatibility with existing and next‑generation AI accelerators.
• Design, implement, and maintain CI/CD pipelines to automate the software release process.
• Collaborate with development teams to integrate new software releases.
• Manage and automate infrastructure provisioning; ensure high availability and scalability of systems through effective infrastructure management.
• Implement monitoring solutions to track system performance; identify bottlenecks and optimize performance.
• Implement security best practices in the DevOps pipeline; conduct regular vulnerability assessments and risk management.

Skills

Software development

System design

Performance optimization

Debugging

Cloud technologies

Education

Bachelor's degree in computer science or equivalent

Job description

Annapurna Labs (U.S.) Inc. in Cupertino, CA is seeking a Software Engineer to lead the development of services that optimize, analyze, and release ML workloads and artifacts. You will drive distributed systems initiatives from architecture through delivery to customers.

The role requires deep knowledge of optimization, resource management, scheduling, and experience with EC2/EKS/Lambda or equivalent cloud services. You will work in a startup‑like environment with cross‑functional collaboration.