Full Job Description
Key job responsibilities
- Collaborate with cross-disciplinary teams that may include hardware design, firmware, compiler, runtime, and ML engineers to develop software that enables and optimizes custom ML accelerator silicon
- Design, develop, and test software across the accelerator stack, including compiler toolchains, device drivers, runtime libraries, profiling tools, simulation frameworks, or chip validation infrastructure
- Build scalable, high-performance systems supporting chip verification, silicon bring-up, or production ML inference/training workloads
- Write clean, maintainable code (C, C++, Python, or Rust) following best practices for systems-level and performance-critical development
- Develop CI/CD pipelines, test automation, and tooling to accelerate the hardware-software co-development lifecycle
- Participate in code reviews, contribute to technical documentation, and leverage GenAI tools to enhance development productivity
- Demonstrate operational excellence through monitoring, debugging, and resolving issues across hardware-software integration boundaries, including on-call responsibilities
- Stay current with emerging technologies in ML accelerators, compiler optimization, computer architecture, and cloud-scale distributed systems
A day in the life
As an Amazonian, you will be matched to a manager and a team, and will have the opportunity to influence the evolution of Amazon technology and work on critical projects early in your career. In addition to working on impactful projects, you will have the opportunity to engage with Amazonians for both personal and professional development, expand your network, and participate in various activities. We give you the tools to own your projects and learn in a real-world settings.
BASIC QUALIFICATIONS
- Are enrolled in a Master's degree in Computer Science, Computer Engineering, or related fields at time of application
- Experience programming with C++ using OOP
- Coursework in Computer Architecture
PREFERRED QUALIFICATIONS
- Coursework, research, or project involving architectural simulators (gem5, QEMU, self-built simulator, etc.)
- Experience with the full software development life cycle, including coding standards, code reviews, source control management, build processes, and testing
- Internship or research experience in silicon, firmware, architecture, or performance analysis
- Experience with SoC, CPU, GPU, and/or ASIC architecture and micro-architecture
- Familiarity with modern C++ (17, 20, etc.)
- Experience with PyTest or GoogleTest
The starting pay for this position is listed below. Starting Day 1 of employment, Amazon offers EAP, Mental Health Support, Medical Advice Line, 401(k) matching. Learn more about our benefits at https://hiring.amazon.com/why-amazon/benefits.
Cupertino, CA, USA - 148,400.00 USD Annually