Staff Engineer - AI Workload Benchmarking

Leadstack Inc

Location not stated

Staff / Principal

3 months ago

AI
System Design
AI/ML
PyTorch
vLLM
TensorRT
FAISS
Milvus
Linux
eBPF
Python
C++

3 months ago

Not scoredNo CV on file. Upload one and this job gets a score out of 100.Upload CV

LeadStack Inc. is an award-winning, one of the nation's fastest-growing, certified minority-owned (MBE) staffing services provider of contingent workforce. As a recognized industry leader in contingent workforce solutions and Certified as a Great Place to Work, we're proud to partner with some of the most admired Fortune 500 brands in the world.

Job Title:Staff Engineer - AI Workload Benchmarking
Duration:6+ Months
Location:San Jose, CA

PR: $65.00 - $75.00/hr on W2
Job Description
• 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.

Looking forward to connecting!

To know more about current opportunities at LeadStack, please visit us athttps://leadstackinc.com/careers/
Should you have any questions, feel free to call me on or send an email.

Staff Engineer - AI Workload Benchmarking · Leadstack Inc

Auto apply with Likeremote