27 days ago
Cupertino, CA, USAMid Level
H1B sponsor
Base Salary
$165k - $224k/yr
Responsibilities
- Design, implement, test, deploy, and maintain software solutions for the AWS Neuron compiler.
- Develop compiler optimization and verification passes for ML workloads running on AWS Inferentia and Trainium.
- Build features that expose AWS accelerator capabilities and improve compiler performance, stability, and usability.
- Create tools to analyze numerical errors and resolve the root causes of compiler defects.
- Collaborate with chip architects, runtime and operating-system engineers, scientists, ML application teams, internal stakeholders, and open-source communities.
- Participate in pre-silicon design, technical design discussions, code reviews, and product feature delivery.
Requirements
- 3+ years of non-internship professional software development experience.
- 2+ years of non-internship experience designing or architecting new and existing systems, including design patterns, reliability, and scaling.
- Experience programming with at least one software programming language.
- Preferred: Master's degree or PhD in Computer Science or a related technical field.
- Preferred: 3+ years of production-grade coding experience in object-oriented languages such as C++ or Java.
- Preferred: experience with compiler design for CPU, GPU, vector engines, or ML accelerators.
- Preferred: experience with open-source compiler toolsets such as LLVM and MLIR.
- Preferred: experience with PyTorch, OpenXLA, StableHLO, JAX, TVM, deep learning models, and algorithms.
- Preferred: experience with modern build systems such as Bazel or CMake.
Benefits
- Comprehensive health insurance, including medical, dental, vision, prescription, life, and AD&D coverage.
- Employee assistance, mental health support, medical advice line, flexible spending accounts, and adoption and surrogacy reimbursement coverage.
- 401(k) matching, paid time off, and parental leave.
- Sign-on payments and restricted stock units may be included in the Amazon package.
- Location: Cupertino, California, USA.
Categories
About Amazon
Amazon builds and operates a global e-commerce marketplace, logistics network, and consumer devices, and provides cloud computing via AWS for businesses and developers. The company earns revenue from online retail, third‑party seller services, subscriptions like Prime, advertising, and AWS usage. Founded in 1994 and headquartered in Seattle, it is publicly traded on NASDAQ (AMZN) and serves customers in dozens of countries.