Sr. SDE, MLA hardware/software co-design, Annapurna Labs Machine Learning Acceleration

About the role

Custom machine learning chips are at the heart of our Trainium machine learning instances, and we invite you to build them with us!

Responsibilities

• Drive the pre-silicon hardware/software co-development for our machine learning chips.
• Work with architecture, design, and emulation teams to build and test our next generation AI chips.
• Write bare-metal software and ML workloads to verify the end-to-end functionality and performance of the SoC.

Requirements

• 5+ years of experience in software development
• Knowledge of HW/SW interfaces and computer architecture
• Proficiency in programming in C/C++, scripting in Bash/Python
• Proficiency in data structures and algorithms

Qualifications

• Knowledge in low level software such as firmware and device drivers
• Knowledge in SoC architecture
• Knowledge in IO (PCIe, AXI), Memory (HBM, DDR), CPU architecture and Interconnects.

Benefits

Final compensation will be determined based on factors including experience, qualifications, and location. Amazon also offers comprehensive benefits including health insurance, 401(k) matching, paid time off, and parental leave. Learn more about our benefits at https://amazon.jobs/en/benefits.