Sr. SDE, MLA hardware/software co-design, Annapurna Labs Machine Learning Acceleration; AWS
Listed on 2026-07-18
-
Software Development
Machine Learning/ ML Engineer, Embedded Software Engineer, AI Engineer (Applied/Software)
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!
BasicQualifications
- 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
- 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 – $ – $ USD annually
- USA, TX, Austin – $ – $ USD annually
- USA, TX, Bee Cave – $ – $ USD annually
(If this job is in fact in your jurisdiction, then you may be using a Proxy or VPN to access this site, and to progress further, you should change your connectivity to another mobile device or PC).