Home / Engineering / Member of Technical Staff, TPU Kernel Engineering

Member of Technical Staff, TPU Kernel Engineering

Inferact · Singapore

Member of Technical Staff, TPU Kernel Engineering is open at Inferact in Singapore. The posting gives a salary of SGD 200K - SGD 400K. Posted on Jun 26, 2026, and still on Inferact’s careers page when we last checked, on Nov 14, 2026.

About the role

We're looking for a TPU kernel engineer to make vLLM a first-class inference engine on Google TPUs. 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…

The opening lines of the posting. The rest is on Inferact’s careers page.

Hiring at Inferact

Inferact has 33 open roles right now, 3 of them remote. Most are in Engineering, Marketing & Community and Finance & Trading. The median advertised salary across its roles with a published range is $300K. See all its roles

Similar Engineering jobs

All Engineering jobs
Apply on company site