2 months ago
Cupertino, CA, USAMid Level
H1B sponsor
Base Salary
$165k - $224k/yr
Responsibilities
- Design, implement, test, deploy, and maintain software solutions that improve Neuron compiler performance, stability, and user experience.
- Build compiler optimization and verification passes for machine learning workloads on AWS Inferentia and Trainium accelerators.
- Develop tools to analyze numerical errors, expose accelerator features to developers, and resolve compiler defects.
- Collaborate with chip architects, runtime and operating-system engineers, scientists, ML application teams, internal teams, customers, and open-source communities.
- Participate in pre-silicon design, design discussions, code reviews, and delivery of new products and features.
Requirements
- At least 3 years of non-internship professional software development experience.
- At least 2 years of non-internship experience designing or architecting new and existing systems, including design patterns, reliability, and scaling.
- Programming experience in at least one software programming language.
- Preferred: a master's degree or PhD in computer science or a related technical field.
- Preferred: at least 3 years of production-grade programming 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
- Health insurance including medical, dental, vision, prescription, Basic Life and AD&D insurance, and optional supplemental life plans.
- Employee assistance and mental health support, a 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 compensation package.
- The position is based in Cupertino, California.
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.