7 days ago
Cupertino, CA, USAMid Level
H1B sponsor
Base Salary
$165k - $224k/yr
Responsibilities
- Design, implement, test, deploy, and maintain software solutions that improve AWS Neuron compiler performance, stability, and usability.
- Build compiler optimization and verification passes and develop features for AWS Inferentia and Trainium accelerators.
- Transform machine-learning models into optimized implementations for deployment on AWS accelerator-based servers.
- Analyze numerical errors and diagnose and resolve compiler defects.
- Collaborate with chip architects, runtime and operating-system engineers, scientists, ML application teams, internal stakeholders, customers, and open-source communities.
- Participate in pre-silicon design, technical design discussions, code reviews, and product feature development.
Requirements
- At least 3 years of non-internship professional software development experience.
- At least 2 years of experience designing or architecting new and existing systems, including design patterns, reliability, and scaling.
- At least 3 years of programming in a modern language such as Java, C++, or C#, including object-oriented design experience.
- Preferred: a master’s degree or equivalent, or a doctorate/PhD or equivalent, in computer science, computer engineering, or a related field.
- Preferred: compiler design experience 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 benefits including medical, dental, vision, prescription, life and AD&D insurance, employee assistance and mental-health support, flexible spending accounts, and adoption and surrogacy reimbursement.
- 401(k) matching, paid time off, and parental leave.
- Sign-on payments and restricted stock units may be included in the compensation package.
- The role is based in Cupertino, California.
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.