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

Amazon; Austin (TX); On-site; USD 159,200 - 215,300; Full time

Job summary

Amazon is seeking a Sr. SDE for MLA hardware/software co-design in Austin, TX. You will drive pre-silicon co-development, writing bare-metal software and ML workloads, and collaborating with architecture, design and emulation teams to test next-generation AI chips.

You should have 5+ years of software development experience, deep HW/SW interface knowledge, and strong C/C++ with scripting skills. This role offers competitive base pay in the Austin market.

Qualifications

• 5+ years of software development experience.
• Experience with HW/SW interfaces and computer architecture.
• Proficiency in C/C++ and scripting languages.
• Strong knowledge of data structures and algorithms.

Responsibilities

• Drive pre-silicon hardware/software co-design for MLA chips.
• Collaborate with architecture, design and emulation teams.
• Write bare-metal software and ML workloads to verify SoC functionality.
• Test and validate end-to-end performance of the chips.

Skills

C/C++

Bash/Python scripting

Data structures

Computer architecture

HW/SW interfaces

Job description

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

AWS Utility Computing (UC) provides product innovations — from foundational services such as Amazon’s Simple Storage Service (S3) and Amazon Elastic Compute Cloud (EC2), to consistently released new product innovations that continue to set AWS’s services and features apart in the industry. As a member of the UC organization, you’ll support the development and management of Compute, Database, Storage, Internet of Things (IoT), Platform, and Productivity Apps services in AWS, including support for customers who require specialized security solutions for their cloud services.

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

As a senior SDE in the pre‑silicon team, you will be responsible for driving the pre‑silicon hardware/software co‑development for our machine learning chips. You will work with architecture, design and emulation teams to build and test our next generation AI chips. You will write bare‑metal software and ML workloads to verify the end‑to‑end functionality and performance of the SoC. If you have a solid background in embedded software, system development or chip design, come join us!

Basic Qualifications

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

Preferred 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

Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status.

Base salary ranges:

• USA, CA, Cupertino – $183,000.00 – $247,600.00 USD annually
• USA, TX, Austin – $159,200.00 – $215,300.00 USD annually
• USA, TX, Bee Cave – $159,200.00 – $215,300.00 USD annually