Senior AI SoC Modeling Engineer, Annapurna Labs Machine Learning Accelerators, AWS

Annapurna Labs

San Diego (CA)

On-site

USD 168,000 - 227,000

Full time

12 hours ago
Be an early applicant

A complete application in a minute

Tailored resume and cover letter, ready to send.

Get past ATS filters

Benefits offered by this job

Health insurance
401(k) matching

Job summary

Annapurna Labs (U.S.) Inc. in California seeks a modeling engineer to develop high-fidelity C++/SystemC models of AWS AI accelerators and SoCs.

You will translate architecture specs into accurate models and validate them against RTL/silicon, working with DV teams across pre-silicon validation environments. You will collaborate with architects, RTL designers, ML software engineers, and compiler teams to assess tradeoffs and drive architectural decisions, contributing to cycle-approximate

Qualifications

  • Experience building functional or performance models of SoCs, ASICs, GPUs, CPUs, or IP blocks.
  • Translate architecture specs or reference implementations into C++ or SystemC models.
  • Experience correlating models against RTL or silicon; DV teams desired.

Responsibilities

  • Develop and maintain high-fidelity functional models of AI/ML accelerators and SoC subsystems.
  • Validate model behavior against RTL simulations, emulation platforms, or silicon measurements; debug discrepancies.
  • Collaborate with design verification teams to integrate models into pre-silicon validation environments.
  • Contribute to cycle-approximate performance modeling and evaluate architecture tradeoffs across compute and memory subsystems.

Skills

C++ modeling
SystemC modeling
SoCs modeling
DV teams experience

Job description

Description

AWS's Trainium and Inferentia chips power the world's largest machine learning clusters. Our team builds C++ models of these custom SoCs that RTL designers, verification engineers, and software teams depend on throughout the silicon development lifecycle. We're looking for a modeling engineer to build and own models that directly impact how our chips are designed, verified, and brought to production.

Description

AWS's Trainium and Inferentia chips power the world's largest machine learning clusters. Our team builds C++ models of these custom SoCs that RTL designers, verification engineers, and software teams depend on throughout the silicon development lifecycle. We're looking for a modeling engineer to build and own models that directly impact how our chips are designed, verified, and brought to production.

What You’ll Do
  • 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
Why This Role Is Interesting
  • Your models are used to verify silicon before it's built — bugs you catch save months of schedule and millions of dollars
  • You'll work at the intersection of software engineering and chip design, with deep visibility into how custom ML accelerators are architected
  • As the team scales, there's a clear path into architectural modeling — using your models to influence chip design decisions, not just validate them
  • Small team, high ownership, direct impact on AWS's most strategic silicon programs
You Will Thrive In This Role If You
  • Have built functional or performance models of SoCs, ASICs, GPUs, CPUs, or IP blocks
  • Are comfortable working with architectural / design specifications or reference implementations and translating them into C++ or SystemC models
  • Understand verification concepts and have worked with DV teams or in pre-silicon validation environments
  • Care about model fidelity and have experience correlating models against RTL or silicon
  • Are interested in expanding into architectural performance modeling as the team grows
  • Enjoy working on a small, high-impact team where you own significant pieces of the stack

No ML background needed. You'll learn the ML accelerator domain on the job.

This role can be based in Cupertino, CA or Austin, TX.

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

Los Angeles County applicants: Job duties for this position include: work safely and cooperatively with other employees, supervisors, and staff; adhere to standards of excellence despite stressful conditions; communicate effectively and respectfully with employees, supervisors, and staff to ensure exceptional customer service; and follow all federal, state, and local laws and Company policies. Criminal history may have a direct, adverse, and negative relationship with some of the material job duties of this position. These include the duties and responsibilities listed above, as well as the abilities to adhere to company policies, exercise sound judgment, effectively manage stress and work safely and respectfully with others, exhibit trustworthiness and professionalism, and safeguard business operations and the Company’s reputation. Pursuant to the Los Angeles County Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.

Our inclusive culture empowers Amazonians to deliver the best results for our customers. If you have a disability and need a workplace accommodation or adjustment during the application and hiring process, including support for the interview or onboarding process, please visit https://amazon.jobs/content/en/how-we-hire/accommodations for more information. If the country/region you’re applying in isn’t listed, please contact your Recruiting Partner.

Amazon also offers comprehensive benefits including 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, and parental leave. Learn more about our benefits at https://amazon.jobs/en/benefits.
USA, CA, Cupertino - 193,300.00 - 261,500.00 USD annually
USA, CA, San Diego - 168,100.00 - 227,400.00 USD annually
USA, TX, Austin - 168,100.00 - 227,400.00 USD annually

Company

- Annapurna Labs (U.S.) Inc.
Job ID: A10375238

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

Senior AI SoC Modeling Engineer, Annapurna Labs Machine Learning Accelerators, AWS
Senior AI SoC Modeling Engineer, Annapurna Labs Machine Learning Accelerators, AWS

Amazon Web Services (AWS) • Austin (TX)

On-site
USD 168,000 - 227,000
Senior AI SoC Modeling Engineer, Annapurna Labs Machine Learning Accelerators, AWS
Senior AI SoC Modeling Engineer, Annapurna Labs Machine Learning Accelerators, AWS

Amazon Web Services (AWS) • Cupertino (CA)

On-site
USD 193,000 - 262,000
Health insurance
401(k) matching
Paid time off
+1
AI SoC Modeling Engineer, Annapurna Labs Machine Learning Accelerators, AWS
AI SoC Modeling Engineer, Annapurna Labs Machine Learning Accelerators, AWS

Amazon Web Services (AWS) • Austin (TX)

On-site
USD 144,000 - 194,000
Health insurance
401(k) matching
Paid time off
+2
AI SoC Modeling Engineer, Annapurna Labs Machine Learning Accelerators, AWS
AI SoC Modeling Engineer, Annapurna Labs Machine Learning Accelerators, AWS

Amazon Web Services (AWS) • Cupertino (CA)

On-site
USD 165,000 - 224,000
Health insurance
RSUs/Stock grants
Paid time off
Senior Architecture Modeling Engineer, AWS Machine Learning Accelerators
Senior Architecture Modeling Engineer, AWS Machine Learning Accelerators

Amazon Web Services (AWS) • Cupertino (CA)

On-site
USD 193,000 - 262,000
Virtual Platform Software Engineer, Annapurna Labs Machine Learning Accelerators, AWS
Virtual Platform Software Engineer, Annapurna Labs Machine Learning Accelerators, AWS

Amazon Web Services (AWS) • Austin (TX)

On-site
USD 144,000 - 194,000
Senior Software Development Engineer, AWS Machine Learning Accelerators
Senior Software Development Engineer, AWS Machine Learning Accelerators

Amazon Web Services (AWS) • Cupertino (CA)

On-site
USD 193,000 - 262,000
Senior Virtual Platform Software Engineer, Annapurna Labs Machine Learning Accelerators, AWS
Senior Virtual Platform Software Engineer, Annapurna Labs Machine Learning Accelerators, AWS

Amazon Web Services (AWS) • Austin (TX)

On-site
USD 168,000 - 227,000
Health insurance
RSUs
401(k) matching
+2
SoC Systems Software Engineer, Annapurna Labs Machine Learning Accelerators, AWS
SoC Systems Software Engineer, Annapurna Labs Machine Learning Accelerators, AWS

Amazon Web Services (AWS) • Austin (TX)

On-site
USD 144,000 - 194,000
Health insurance
401(k) matching
Paid time off
+1
Senior Virtual Platform Software Engineer, Annapurna Labs Machine Learning Accelerators, AWS
Senior Virtual Platform Software Engineer, Annapurna Labs Machine Learning Accelerators, AWS

Amazon Web Services (AWS) • Cupertino (CA)

On-site
USD 193,000 - 262,000