Sr. Sde, Mla Hardware/software Co-design, Annapurna Labs Machine Learning Acceleration

Amazon Amazon · Big Tech · Austin, TX · Software Development

Senior Software Development Engineer focused on pre-silicon hardware/software co-development for next-generation machine learning chips (like Trainium) used in AWS. The role involves working with architecture, design, and emulation teams, writing bare-metal software and ML workloads to verify chip functionality and performance.

What you'd actually do

  1. driving the pre-silicon hardware/software co-development for our machine learning chips
  2. work with architecture, design and emulation teams to build and test our next generation AI chips
  3. write bare-metal software and ML workloads to verify the end-to-end functionality and performance of the SoC

Skills

Required

  • 5+ YoE 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

Nice to have

  • 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

What the JD emphasized

  • pre-silicon hardware/software co-development
  • next generation AI chips
  • bare-metal software and ML workloads

Other signals

  • machine learning chips
  • Trainium machine learning instances
  • pre-silicon hardware/software co-development
  • next generation AI chips
  • bare-metal software and ML workloads