Member of Technical Staff, TPU Kernel Engineering
Inferact is hiring for Member of Technical Staff, TPU Kernel Engineering in San Francisco. Staff level. First seen August 6, 2026; open as of November 11, 2026. Jobs taken straight from company career pages — no aggregator in between.
With Destava: How well do you fit? · Tailor my résumé · Write a cover letter
- Company
- Inferact
- Where
- San Francisco
- Level
- Staff
- First seen
- August 6, 2026
- Listed via
- Ashby
What the posting says
A large part of this role is designing, writing, and optimizing kernels for TPU in Pallas and JAX, for the operations that dominate LLM inference: attention, GEMM, sampling, KV cache, and fused operations. You'll work at the boundary of kernels, compilers, inference systems, and hardware architecture. You'll profile production-relevant models on TPU, find the bottleneck, and write the kernel or compiler change that removes it, with clear correctness, latency, and throughput benchmarks. Your work will help make TPU support in vLLM usable, fast, benchmarked, and maintainable. Inferact has…
Read more and apply on Inferact's site
More jobs like this
A search agent finds the jobs this list misses: the ones that mean the same without the same words. Have Destava watch for member of technical staff, tpu kernel engineering jobs in San Francisco, new matches every morning on your own account.