Amazon Web Services (AWS) · 20 hours ago
Machine Learning Compiler Engineer, Annapurna Labs
Cupertino, CA
Full-time
Onsite
Mid Level
$129K/yr - $224K/yr
3+ years expAmazon Web Services (AWS) is seeking skilled compiler engineers to join their AWS Neuron Compiler team. The role involves designing, developing, and optimizing features for a deep learning compiler stack, collaborating with cross-functional teams to enhance system performance and tackle technical challenges.
Agentic AIConsultingDevOpsInformation TechnologySoftwareWeb Development
Responsibilities
Solve challenging technical problems, often ones not solved before, at every layer of the stack
Design, implement, test, deploy and maintain innovative software solutions to transform service performance, durability, cost, and security
Research implementations that deliver the best possible experiences for customers
Build high-impact solutions to deliver to our large customer base
Participate in design discussions, code review, and communicate with internal and external stakeholders
Work cross-functionally to help drive business decisions with your technical input
Work in a startup-like development environment, where you’re always working on the most important stuff
Qualification
Required
B.S. or M.S. in computer science or related field
Proficiency with 1 or more of the following programming languages: C++ (preferred), Python
3+ years of non-internship professional software development experience
2+ years of experience developing compiler optimization, graph-theory, hardware bring-up, FPGA placement and routing algorithms, or hardware resource management
Preferred
M.S. or Ph.D. in computer science or related field
Strong knowledge in one or more of the areas of: compiler design, instruction scheduling, memory allocation, data transfer optimization, graph partitioning, parallel programing, code generation, Instruction Set Architectures, new hardware bring-up, and hardware-software co-design
Experience with LLVM and/or MLIR
Experience with developing algorithms for simulation tools
Experience is TensorFlow, PyTorch, and/or JAX
Experience in LLM, Vision or other deep-learning models
Benefits
Equity
Sign-on payments
Full range of medical, financial, and/or other benefits
Company
Amazon Web Services (AWS)
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.
10001+ employees
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 StageTotal Funding
unknownKey Investors
BIRD Foundation
2025-01-22Grant
Leadership Team
Matt Garman
Chief Executive Officer
Anand Desikan
CTO, CXO Advisor, and Enterprise Technologist
Recent News
2026-01-09
2026-01-09
2026-01-09
Company data provided by crunchbase