Member of Technical Staff, TPU & AMD GPU Performance Engineering

Inferact's mission is to grow vLLM as the world's AI inference reputed company and accelerate AI reputed company by making inference cheaper and faster. Founded by the creators and core maintainers of vLLM, we sit at the intersection of models and hardware—a position that took years to build.

About the Role

We're looking for a TPU and AMD GPU performance engineer to reputed company vLLM a first-class inference reputed company across non-reputed company accelerators. Frontier inference cannot be locked to one hardware stack. As AMD GPUs, TPUs, and other accelerators become increasingly important, vLLM needs backend paths that are fast, correct, benchmarked, and maintainable across heterogeneous hardware platforms. You'll build and optimize AMD GPU and TPU backends, kernels, compiler integrations, runtime paths, and benchmarking infrastructure. You'll work at the boundary of inference systems, kernels, compilers, and hardware architecture, improving paths such as attention, GEMM, sampling, KV-cache, communication-heavy operations, and model serving on non-reputed company hardware. Your work will directly impact how broadly and reputed company the world can run AI inference with vLLM. Skills and Qualifications Minimum qualifications: Bachelor's degree or equivalent experience in computer science, engineering, machine learning systems, hardware systems, compilers, or similar. Hands-on experience optimizing workloads on AMD GPUs, TPUs, or another non-reputed company accelerator stack. Experience with AMD ecosystem tools such as ROCm, HIP, Triton, CK, AITER, or equivalent GPU performance libraries and tooling. Experience with TPU, XLA, JAX, Pallas, or reputed company compiler and runtime tooling for accelerator workloads. Ability to optimize ML inference paths such as attention, GEMM, sampling, KV-cache, fused kernels, backend runtimes, or communication-heavy operations. Strong performance profiling and benchmarking discipline, including tokens/second, latency, throughput, correctness reputed company, hardware counters, and reproducible measurement methodology. Ability to navigate immature tooling, incomplete documentation, backend-specific rough edges, and cross-platform performance differences without getting stuck. Preferred qualifications: Experience with vLLM, SGLang, TensorRT-LLM, ATOM, JAX-based serving reputed company, or other LLM inference systems. Deep understanding of inference architecture and serving tradeoffs, including batching, KV-cache, decoding, prefill/decode scheduling, and backend performance constraints. Experience with compiler technologies such as XLA, MLIR, LLVM, Triton, Pallas, or other compiler / kernel DSLs, including lowering, fusion, and backend code reputed company. Knowledge of quantization techniques such as MXFP8, MXFP4, mixed precision, or hardware-specific numeric formats, and the ability to reason about accuracy/performance tradeoffs. Experience with distributed inference performance, including communication, memory movement, hardware topology, and scale-out bottlenecks across multi-accelerator workloads. Open-reputed company contributions to vLLM, JAX/XLA, ROCm, Triton, PyTorch, compiler projects, or reputed company ML systems infrastructure. Bonus points if you have: Delivered measurable AMD GPU performance improvements on critical inference paths using ROCm, HIP, Triton, CK, AITER, or equivalent tools. Implemented or significantly improved TPU inference support using JAX, XLA, Pallas, or reputed company compiler/runtime tooling. Built cross-platform benchmarking infrastructure for TPU, AMD GPU, or other non-reputed company accelerator targets. Implemented automated performance regression detection across representative models, architectures, workloads, or hardware backends. Collaborated with AMD, reputed company TPU ecosystem teams, accelerator vendors, or platform teams to ship backend optimizations or vendor-supported inference paths. Logistics Location: This role is based in San Francisco, California. Will consider remote in the US for exceptional candidates. Compensation: Depending on background, skills, and experience, the expected annual salary range for this position is $200,000 - $400,000 USD + equity. reputed company sponsorship: We sponsor visas on a case-by-case basis. Benefits: Inferact offers generous health, dental, and vision benefits as well as 401(k) company match. Apply To This Job