Sr. Machine Learning - Compiler Engineer III, AWS Neuron, Annapurna Labs jobs in United States
cer-icon
Apply on Employer Site
company-logo

Amazon Web Services (AWS) · 6 hours ago

Sr. Machine Learning - Compiler Engineer III, AWS Neuron, Annapurna Labs

positionCupertino, CA
timeFull-time
remoteOnsite
senioritySenior Level
money$151K/yr - $262K/yr
date5+ years exp

Amazon Web Services (AWS) is at the forefront of innovation in machine learning accelerators, and they are seeking a Senior Machine Learning Compiler Engineer to join their Annapurna Labs team. In this role, you will develop and scale a compiler for large ML workloads, architect features, and collaborate with AWS ML services teams.

Agentic AIConsultingDevOpsInformation TechnologySoftwareWeb Development
check
H1B Sponsor Likelynote

Responsibilities

Supporting the ground-up development and scaling of a compiler to handle the world's largest ML workloads
Architecting and implementing business-critical features
Publish cutting-edge research
Contributing to a brilliant team of experienced engineers
Leverage your technical communications skill as a hands-on partner to AWS ML services teams
Involved in pre-silicon design
Bringing new products/features to market
Many other exciting projects

Qualification

Compiler developmentMachine LearningC++PythonTensorFlowPyTorchJAXResource managementCode generationCompute graph optimizationTechnical communicationTeam collaboration

Required

5+ years of leading design or architecture (design patterns, reliability and scaling) of new and existing systems experience
2+ years of experience in developing compiler features and optimizations
Proficiency with 1 or more of the following programming languages: C++, C, Python

Preferred

Master or PhD degree in computer science or equivalent
Proficiency with resource management, scheduling, code generation, and compute graph optimization
Experience optimizing Tensorflow, PyTorch or JAX deep learning models
Experience with multiple toolchains and Instruction Set Architectures

Benefits

Equity
Sign-on payments
Full range of medical, financial, and/or other benefits

Company

Amazon Web Services (AWS)

twittertwittertwitter
Glassdoor3.8
company-logo
Launched in 2006, Amazon Web Services (AWS) began exposing key infrastructure services to businesses in the form of web services -- now widely known as cloud computing.
dateFounded in 2002
location
Seattle, Washington, USA
people-size10001+ employees
web-link
http://aws.amazon.com

H1B Sponsorship

Amazon Web Services (AWS) has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (22803)
2024 (21175)
2023 (19057)
2022 (24088)
2021 (12233)
2020 (14881)

Funding

Current Stage
Late Stage
Total Funding
unknown
Key Investors
BIRD Foundation
2025-01-22Grant

Leadership Team

leader-logo
Matt Garman
Chief Executive Officer
linkedin
leader-logo
Anand Desikan
CTO, CXO Advisor, and Enterprise Technologist
linkedin

Recent News

Tech In Africa
Data Sovereignty: The Push for African-Hosted Cloud Infrastructure
2026-01-14
Irish Tech News
Europe’s AI problem isn’t technology but policy
2026-01-14
EIN Presswire
Options For All Launches First-of-Its-Kind Cloud Developer Program Powered by Amazon Web Services (AWS) Cloud Institute
2026-01-13
Company data provided by crunchbase