2 months ago
Base Salary
$193k - $262k/yr
Responsibilities
- Design and implement high-performance compute kernels for machine-learning operations using Neuron architectures and programming models.
- Analyze and optimize kernel-level performance across multiple generations of Neuron hardware.
- Use profiling tools to identify and resolve performance bottlenecks.
- Implement compiler optimizations including fusion, sharding, tiling, and scheduling.
- Work directly with customers to enable and optimize machine-learning models on AWS accelerators.
- Collaborate with compiler, runtime, framework, and hardware teams on kernel optimization techniques.
- Architect and implement business-critical features, publish research, and mentor experienced engineers.
Requirements
- 5+ years of non-internship professional software development experience.
- 5+ years of programming experience with at least one software programming language.
- 5+ years of experience leading the design or architecture of new and existing systems.
- 5+ years of full software development lifecycle experience, including coding standards, code reviews, source control management, build processes, testing, and operations.
- Experience as a mentor, tech lead, or engineering team lead.
- Preferred: bachelor’s degree in computer science or equivalent.
- Preferred: 6+ years of full software development experience.
- Preferred: expertise in ML or HPC accelerator architectures, GPU kernel optimization, GPGPU computing, high-performance libraries, low-level GPU performance optimization, parallel programming, and GPU memory hierarchies.
- Preferred: experience with CUDA, NKI, Triton, OpenCL, SYCL, ROCm, NVIDIA PTX, AMD GPU ISA, LLVM/MLIR backend development, PyTorch, TensorFlow, and GPU backends.
Benefits
- Health insurance including medical, dental, vision, prescription, Basic Life and AD&D insurance, and supplemental life plan options.
- Employee assistance and mental health support, medical advice line, and flexible spending accounts.
- 401(k) matching, paid time off, and parental leave.
- Adoption and surrogacy reimbursement coverage.
- The team offers flexibility in working hours and emphasizes work-life balance.
- Position is based in Cupertino, California.
- Base salary range is $193,300.00–$261,500.00 USD annually; sign-on payments and restricted stock units may also be included.
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.