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) ยท 2 weeks 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. The role involves developing a deep learning compiler stack to handle large ML workloads and collaborating with AWS ML services teams to bring new products and features to market.

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 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++TensorFlow optimizationPyTorch optimizationJAX optimizationResource 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

EIN Presswire
Options For All Launches First-of-Its-Kind Cloud Developer Program Powered by Amazon Web Services (AWS) Cloud Institute
2026-01-13
Newswire
Aible Helps Retailers Navigate Volatility And Boost Business Outcomes with AI Agents That Explain What's Changing & Why Across the Business
2026-01-13
Small Business Trends
35 Startups Chosen for 2026 Cybersecurity Accelerator with AWS and NVIDIA
2026-01-12
Company data provided by crunchbase