Software Development Manager, ML Accelerators, AWS Neuron, Annapurna Labs

Amazon Web Services (AWS)

Seattle (WA)

On-site

USD 180,000 - 260,000

Full time

6 days ago

Be an early applicant

Get more replies from employers

Send a job-specific resume in minutes.

Benefits offered by this job

Health insurance

Medical

Dental

Vision

Prescription

Basic life & AD&D insurance

Supplemental life plans

EAP

Mental health support

Medical advice line

Flexible spending accounts

Adoption and surrogacy reimbursement

401(k) matching

Paid time off

Parental leave

Restricted stock units

Sign-on payments

Job summary

Amazon Web Services (AWS) in Seattle leads a team of compiler engineers to design and develop machine learning compiler solutions for AWS Neuron. You will drive advanced optimization techniques, guide hardware bring-up for next‑generation chips, and influence pre-silicon design decisions.

You will mentor engineers, collaborate with research and product teams, and deliver scalable compiler infrastructure that accelerates ML workloads across AWS.

Qualifications

Bachelor's degree in a relevant field.
At least 5 years of software engineering experience.
3 years of management experience.
Proficiency in computer architecture, code generation, and hardware bring-up.

Skills

Software engineering management

Team leadership

Mentorship

Machine learning

Optimization techniques

Computer architecture

Code generation

Hardware bring-up

Chip architecture

JAX

PyTorch

TensorFlow

LLVM

MLIR

Education

Bachelor's degree

Tools

LLVM

MLIR

PyTorch

JAX

TensorFlow

Job description

You will lead a team of compiler engineers to design and develop machine learning compiler solutions for AWS Neuron. This role involves driving advanced optimization techniques, managing hardware bring-up for next-generation chips, and influencing pre-silicon design decisions.

Requirements

Candidates must have a bachelor's degree in a relevant field, at least 5 years of software engineering experience, and 3 years of management experience. Proficiency in computer architecture, code generation, and hardware bring-up is required.

Key Skills