Member of Technical Staff, TPU Kernel Engineering
Inferact is hiring for Member of Technical Staff, TPU Kernel Engineering in Singapore. Staff level. First seen August 6, 2026; open as of November 12, 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
- Singapore
- 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
Senior Integration Lead - Advanced Packaging 3DHi and 2.5D
Save to applyAutonomous Vehicle Hardware Intern
Save to applySenior Engineer, HIG-HBM Product Engineering (Media Health Reliability)
Save to applyStaff/Snr Product Engineer (Advance Data Analytics), Heterogeneous Integration Group (HIG), High Bandwidth Memory (HBM) Product Engineering
Save to apply
See all 105 Computer Hardware Engineers jobs in Singapore
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 Singapore, new matches every morning on your own account.
This page includes information from the O*NET 31.0 Database by the U.S. Department of Labor, Employment and Training Administration (USDOL/ETA). Used under the CC BY 4.0 license. O*NET® is a trademark of USDOL/ETA.