← All roles

ML Accelerator Performance Validation Engineer, Post Silicon Validation

Amazon· Software Development
Apply on company site
Hybrid (Austin, TX)$143.7k - $194.4kFull-timePosted 4mo ago

About the role

Join Annapurna Labs (AWS) Post-Silicon Validation to quantify and qualify the performance of AWS's custom ML training chips against architectural targets. Build benchmarks, analyze throughput and memory performance, profile real ML workloads, identify bottlenecks, and drive optimization to ensure silicon meets latency, throughput, and efficiency targets at cloud scale.

Responsibilities

  • Design and execute performance benchmarks spanning micro-architectures to full model training
  • Measure and analyze compute throughput, memory bandwidth, interconnect latency, and more
  • Profile real ML workloads (transformer models, LLMs, vision models) on silicon
  • Identify performance bottlenecks and work with architecture teams on optimization
  • Build automated performance regression dashboards and tracking infrastructure
  • Correlate silicon measurements against RTL simulation and emulation predictions

Qualifications

  • 3+ years of non-internship professional software development experience
  • 2+ years of non-internship design or architecture (design patterns, reliability and scaling) of new and existing systems experience
  • Experience with Machine Learning and Large Language Model fundamentals, including architecture, training/inference lifecycles, and optimization of model execution, or experience working with PyTorch or JAX software
  • Bachelor's degree in computer science, engineering, mathematics or equivalent, or experience in Java, C++, Python, or a related language
  • 3+ years of experience with hardware performance counters and profiling tools for analyzing and optimizing system and application performance
  • Strong understanding of computer architecture fundamentals including memory hierarchies (caches, DRAM, HBM), compute pipelines, and interconnect topologies
  • Experience applying statistical methods, regression analysis, and data visualization techniques to interpret performance data and drive optimization decisions

Skills

Required:Performance benchmarkingProfiling and measurementML workload profiling on siliconBenchmark automation

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
  • parential leave
Last updated Oct 25, 2026. Interested? Apply on the company site.

About Amazon

Amazon.com, Inc. or its affiliates

liquid-ai-recruiting-target
Company profileWebsite

More open roles at Amazon