22 hours 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
