AI SoC Modeling Engineer
Amazon· Annapurna Labs Machine Learning Accelerators, AWS
Hybrid (Cupertino, CA; Austin, TX)$165.2k - $223.6kFull-timePosted 1mo ago
About the role
AWS's Trainium and Inferentia chips power the world's largest machine learning clusters. The team builds C++ models of these custom SoCs used by RTL designers, verification engineers, and software teams throughout the silicon development lifecycle. The role focuses on building and owning high-fidelity models of AI/ML accelerators and their SoC subsystems to influence design, verification, and production.
Responsibilities
- Develop and maintain high-fidelity functional model of AI/ML accelerator and its SoC subsystems, including compute engines, memory hierarchies, on-chip interconnects, and data paths — translating architecture specs and RTL behavior into accurate, testable C++ models
- Validate model behavior against RTL simulations, emulation platforms, or silicon measurements; debug discrepancies and drive model-to-RTL correlation to high fidelity
- Partner with design verification teams to integrate models into pre-silicon validation environments and catch architectural bugs early in the design cycle
- Collaborate with architects/micro-architects, RTL design engineers, ML SW engineers, and compiler engineers to evaluate architecture and microarchitecture tradeoffs and help make hardware design decisions
- Contribute to cycle-approximate performance model effort enabling architectural exploration ahead of RTL availability, early software development
- Quantify system-level tradeoffs across compute, memory bandwidth, networking, and storage to influence reference architectures and long-term silicon strategy
- Build and improve modeling infrastructure: simulation frameworks, regression suites, automated correlation checks, and coverage-driven validation flows
- Develop modeling methodologies and tools that scale across multiple IP blocks and SoC generations, improving team efficiency and model reuse
Qualifications
- Experience programming languages such as C/C++, Python, Java or Perl
- 2+ years writing functional or performance models of hardware (SoCs, ASICs, GPUs, CPUs, IP blocks)
- Familiarity with SoC, CPU, GPU, and/or ASIC architecture and micro-architecture
Skills
Required:C/C++Python
Nice to have:SystemC or TLMRTL correlation / DV collaborationModern C++ (20+)PyTest / GoogleTest or similarMulti-threaded simulation
Benefits
- health insurance (medical, dental, vision, prescription)
- Basic Life & AD&D insurance and option for Supplemental life plans
- EAP, Mental Health Support, Medical Advice Line
- Flexible Spending Accounts
- Adoption and Surrogacy Reimbursement coverage
- 401(k) matching
- paid time off
- parental leave
Last updated Oct 25, 2026. Interested? Apply on the company site.
More open roles at Amazon
- IT Support Associate II, Ops Tech Solutions (OTS)todayPlainfield, Indiana · Operations IT Support Engineering
- Head of Energy and Water, AWS LegaltodayHybrid (Seattle, WA) · Legal
- Software Development Engineer, Creative Asset Library, CreativeXtodaySeattle · Software Development
- Software Development Engineer, Creative Asset Library, CreativeXtodayHybrid (Seattle, WA) · Software Development
- IT Support Associate II, OTStodayIrving, TX · Operations IT Support Engineering