2 months ago
Cupertino, CA, USAMid Level
H1B sponsor
Base Salary
$165k - $224k/yr
Responsibilities
- Design and implement high-performance compute kernels for machine-learning operations using the Neuron architecture 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 their machine-learning models on AWS accelerators.
- Collaborate with compiler, runtime, framework, and hardware teams on kernel optimization techniques.
- Create metrics, implement automation, resolve software defects, participate in design discussions and code reviews, and provide technical input to business decisions.
- Publish research and mentor experienced engineers.
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.
- Experience programming with at least one software programming language.
- Preferred: at least 3 years of full software development life-cycle experience, including coding standards, code reviews, source control management, build processes, testing, and operations.
- Preferred: bachelor’s degree in computer science or equivalent.
Benefits
- Comprehensive health insurance covering medical, dental, vision, prescription, Basic Life and AD&D insurance, and optional supplemental life plans.
- Employee Assistance Program, 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 are included in the overall Amazon package.
- Flexible working hours and an emphasis on work-life balance.
- Inclusive culture with employee-led affinity groups and ongoing learning experiences.
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.