Staff Engineer - AI Workload Benchmakring

Saige PartnersSan Jose, CA
Onsite

About The Position

Align benchmarking insights to leadership in memory (HBM, DRAM, CXL) and storage (SSD/NAND) to inform product roadmaps. Enable next-generation AI infrastructure solutions through performance-driven system design, validation, and standards engagement.

Requirements

  • Bachelor's/ Master's degree in Computer Science, Computer Engineering, Electrical Engineering, or a related field; advanced degree preferred.
  • Professional experience in systems performance engineering, storage performance, or AI infrastructure benchmarking.
  • Demonstrated expertise with NVMe SSDs and storage stack performance analysis (block layer, page cache, file systems, asynchronous I/O).
  • Hands-on experience with AI/ML workloads — LLM training and inference frameworks (PyTorch, vLLM, TensorRT-LLM, or equivalent), embedding pipelines, or vector databases (FAISS, Milvus, DiskANN, HNSW).
  • Strong proficiency with Linux performance and tracing tools: blktrace, perf, eBPF/bpftrace, ftrace, BCC, iostat, fio.
  • Working knowledge of GPU systems and accelerator I/O paths
  • Experience designing and executing benchmarks against industry standards (MLPerf Storage, or equivalent).
  • Proficiency in Python for benchmarking automation, data analysis, and visualization; comfort with C/C++ for systems-level work.
  • Proven ability to deliver structured technical reports, characterization studies, and reproducible benchmark artifacts to a senior engineering audience.

Nice To Haves

  • advanced degree preferred

Responsibilities

  • Align benchmarking insights to leadership in memory (HBM, DRAM, CXL) and storage (SSD/NAND) to inform product roadmaps.
  • Enable next-generation AI infrastructure solutions through performance-driven system design, validation, and standards engagement.

Benefits

  • benefit package
  • convenient weekly payment solutions
© 2026 Teal Labs, Inc
Privacy PolicyTerms of Service