E

Etched

Designs chips and servers for running AI models

MTS – Inference Performance Visibility

Systems Software EngineeringConfirmed live in the last 24 hours
Full-timeSan Jose, CA, USA

About the Job

Requirements

  • Strong proficiency in C++ or Rust.
  • Proficiency in Python is a plus.
  • Deep understanding of computer architecture, including CPUs, GPUs, and accelerators; memory hierarchies such as caches and DRAM; and interconnects, especially PCIe.
  • Proven experience in low-level performance analysis, profiling, and bottleneck identification on complex hardware systems such as GPUs, CPUs, FPGAs, or custom accelerators.
  • Experience with performance analysis tools such as NVIDIA Nsight, AMD uProf, Intel VTune, perf, Tracy, or ETW.
  • Experience working close to hardware, potentially reading performance counters or interacting directly with device drivers.

Responsibilities

  • Lead the design and architecture of a comprehensive performance analysis suite, including data collection mechanisms, data processing pipelines, analysis engines, and user interfaces (CLI and/or GUI).
  • Develop robust methods to capture performance data directly from custom ML accelerator hardware, such as hardware performance counters, execution unit status, and memory access patterns, via driver interfaces or other mechanisms.
  • Implement tracing for host-side API calls, including runtime libraries and driver interactions, and system-level events such as CPU activity, PCIe traffic, memory usage, and network contention related to workloads.
  • Design and implement techniques to accurately correlate performance events across the host CPU, device driver, PCIe bus, multiple accelerators, and multiple hosts, ensuring precise time synchronization.
  • Build analysis modules to automatically interpret collected trace and counter data, identifying key performance limiters such as compute-bound, memory bandwidth-bound, latency-bound, PCIe-bound, and specific hardware bottlenecks.
  • Develop intuitive visualizations, including timelines, dependency graphs, resource utilization charts, and statistical summaries, to communicate performance characteristics and bottlenecks to users.
  • Work closely with hardware architects, firmware engineers, driver developers, compiler engineers, and ML application engineers to understand their needs, define tool requirements, and provide expert guidance on performance analysis and optimization using the tool.
  • Architect and implement the core data collection framework for hardware performance counters on a custom PCIe-based accelerator.
  • Develop a kernel driver module or user-space service for low-overhead tracing of accelerator activity.
  • Design and build a correlated timeline view visualizing CPU API calls, driver submissions, PCIe transfers, and accelerator execution units.
  • Create an analysis pass to detect and quantify memory access inefficiencies or PCIe bandwidth saturation while transacting on a PCIe-attached accelerator.

Desired Qualifications

  • Direct experience developing performance analysis or debugging tools.
  • Experience with ML accelerator architectures such as GPUs or TPUs.
  • Experience with kernel-mode driver development for Linux or Windows.
  • Understanding of compiler internals, code generation, and optimization.
  • In-depth knowledge of the PCIe protocol and PCIe analysis tools.
  • Experience with multi-chip or multi-host accelerator systems such as TPU pods or NVIDIA DGX clusters.
  • Experience with firmware or embedded systems development.
  • Experience with hardware description languages such as Verilog or VHDL, or hardware verification.

About the Company

Etched

Designs chips and servers for running AI models

Etched designs AI chips and the server racks built around them for AI companies and cloud providers, focusing only on inference, the work of running trained models to answer requests. In 2024, it unveiled a chip hard-wired to run only transformers, the type of AI model behind ChatGPT, after betting at its founding that they would dominate AI. Etched develops its chips, racks, and software together, and each rack links many chips through a shared pool of fast memory so very large models with long inputs can answer quickly. It earns money by selling these complete systems rather than individual chips. Etched was founded in 2022 and is based in San Jose, California.

Company size
201-500
Company stage
Series D
Total funding
$1.6B
Headquarters
San Jose, CA
Founded
2022

Recently Posted Jobs

Explore more opportunities at Etched

See more jobs