SIGN IN
ML Compiler Engineer , AWS Neuron, Annapurna Labs jobs in United States
cer-icon
Apply on Employer Site
company-logo

Amazon Web Services (AWS) · 18 hours ago

ML Compiler Engineer , AWS Neuron, Annapurna Labs

positionToronto, Ontario, Canada
timeFull-time
remoteOnsite
seniorityMid, Senior Level
date3+ years exp
Amazon Web Services (AWS) is focused on making deep learning accessible for developers and has created innovative solutions for this purpose. The ML Compiler Engineer will work on the Neuron SDK to optimize the performance of neural net models, architect business-critical features, and mentor a team of engineers while innovating on behalf of customers.
Agentic AIConsultingDevOpsInformation TechnologySoftwareWeb Development

Responsibilities

Develop and scale the compiler to handle the world's largest ML workloads
Support the development of a compiler targeting AWS Inferentia and Trainium
Innovate on behalf of our customers and leverage technical communication skills as a hands-on partner to AWS ML services teams
Be involved in pre-silicon design and bring new products/features to market

Qualification

Compiler designSystem Level performance analysisLLVMDeep learning modelsSoftware development life cycleProgramming languagesPyTorchTechnical communicationMentorshipTeam collaboration

Required

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

Preferred

3+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience
Bachelor's degree in computer science or equivalent
Experience in compiler design for CPU/GPU/Vector engines/ML-accelerators
Experience with System Level performance analysis and optimization
Experience with LLVM and/or MLIR
Experience with the following technologies: PyTorch, OpenXLA, StableHLO, JAX, TVM, deep learning models, and algorithms

Benefits

Inclusive culture
Flexibility in working hours
Mentorship
Career growth

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

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

Digital News Asia
Maxis becomes the first Malaysian telco to migrate mission-critical workloads to the AWS Malaysia region
2026-02-06
CRN
AWS Layoffs Hit US And Canada Employees As Amazon Sheds 16,000 Jobs
2026-02-06
EuropaWire
Prosus Deepens Collaboration with AWS to Drive AI Innovation Across Its Global Portfolio
2026-02-06
Company data provided by crunchbase