Singapore, Singapore · On-site
$200k–$400k/yr
Inferact's mission is to grow vLLM as the world's AI inference engine and accelerate AI progress by making inference cheaper and faster. Founded by the creators and core maintainers of vLLM, we sit at the intersection of…
Skills: TPU Performance Engineering, JAX, XLA, Pallas, ML Kernel Optimization
San Francisco, California, United States · Remote OK
$200k–$400k/yr
Inferact's mission is to grow vLLM as the world's AI inference engine and accelerate AI progress by making inference cheaper and faster. Founded by the creators and core maintainers of vLLM, we sit at the intersection of…
Skills: ROCm, HIP, Triton, CK, AITER
Singapore, Singapore · On-site
$200k–$400k/yr
Inferact's mission is to grow vLLM as the world's AI inference engine and accelerate AI progress by making inference cheaper and faster. Founded by the creators and core maintainers of vLLM, we sit at the intersection of…
Skills: AMD GPU Optimization, TPU Performance Engineering, ROCm, HIP, Triton
Singapore, Singapore · On-site
$200k–$400k/yr
Inferact's mission is to grow vLLM as the world's AI inference engine and accelerate AI progress by making inference cheaper and faster. Founded by the creators and core maintainers of vLLM, we sit at the intersection of…
Skills: CUDA, C++, Python, GPU Architecture, Kernel Optimization
Singapore, Singapore · On-site
$200k–$400k/yr
Inferact's mission is to grow vLLM as the world's AI inference engine and accelerate AI progress by making inference cheaper and faster. Founded by the creators and core maintainers of vLLM, we sit at the intersection of…
Skills: Python, PyTorch, vLLM, TensorRT-LLM, SGLang
Singapore, Singapore · On-site
$200k–$400k/yr
Inferact's mission is to grow vLLM as the world's AI inference engine and accelerate AI progress by making inference cheaper and faster. Founded by the creators and core maintainers of vLLM, we sit at the intersection of…
Skills: Rust, Go, C++, Distributed Systems, Network Protocols
Singapore, Singapore · On-site
$200k–$400k/yr
Inferact's mission is to grow vLLM as the world's AI inference engine and accelerate AI progress by making inference cheaper and faster. Founded by the creators and core maintainers of vLLM, we sit at the intersection of…
Skills: Kubernetes, Container Orchestration, Kubernetes Operators, Python, Rust
San Francisco, California, United States · Remote OK
$200k–$400k/yr
Inferact's mission is to grow vLLM as the world's AI inference engine and accelerate AI progress by making inference cheaper and faster. Founded by the creators and core maintainers of vLLM, we sit at the intersection of…
Skills: Python, PyTorch, vLLM, TensorRT-LLM, SGLang
San Francisco, California, United States · Remote OK
$200k–$400k/yr
Overview Inferact's mission is to grow vLLM as the world's AI inference engine and accelerate AI progress by making inference cheaper and faster. Founded by the creators and core maintainers of vLLM, we sit at the inters…
Skills: LLM Inference Systems, Developer Relations, GPU Serving, Technical Writing, Model Serving
San Francisco, California, United States · Remote OK
$200k–$400k/yr
Inferact's mission is to grow vLLM as the world's AI inference engine and accelerate AI progress by making inference cheaper and faster. Founded by the creators and core maintainers of vLLM, we sit at the intersection of…
Skills: AMD GPU Optimization, TPU Performance Engineering, ROCm, HIP, Triton
Sign up with Clera and we'll reach out the moment a role actually fits you — no more spraying applications into the void.
Build and optimize TPU backends, compiler integrations, and runtime paths to make vLLM a first-class inference engine on Google TPUs. Improve production-relevant model serving by optimizing kernels and benchmarking latency and throughput.
Requires a Bachelor's degree in CS or a related field and hands-on experience optimizing TPU workloads using JAX, XLA, or Pallas. Candidates should have deep knowledge of TPU execution, memory behavior, and ML kernel optimization.
Inferact's mission is to grow vLLM as the world's AI inference engine and accelerate AI progress 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.
We're looking for a TPU performance engineer to make vLLM a first-class inference engine on Google TPUs. You'll build and optimize TPU backends, compiler integrations, runtime paths, and benchmarking infrastructure using JAX, XLA, Pallas, and related tooling so vLLM can deliver frontier inference performance on TPU hardware.
You'll work at the boundary of inference systems, kernels, compilers, and hardware architecture, improving production-relevant model serving on TPU with clear correctness, latency, and throughput benchmarks. Your work will help make TPU support in vLLM usable, fast, benchmarked, and maintainable.
Minimum qualifications:
Bachelor's degree or equivalent experience in computer science, engineering, systems, machine learning, or similar.
Hands-on experience building or optimizing TPU workloads using JAX, XLA, Pallas, or related compiler and runtime tooling.
Deep understanding of TPU execution, memory behavior, compilation, and performance constraints for ML workloads.
Experience optimizing ML kernels or inference paths such as attention, GEMM, sampling, KV cache, fused kernels, or backend runtime paths.
Strong performance profiling and benchmarking skills, with the ability to use measurements, compiler artifacts, correctness tests, and reproducible benchmarks to guide optimization work.
Preferred qualifications:
Experience with vLLM, SGLang, TensorRT-LLM, XLA-based serving, or other LLM inference systems.
Familiarity with batching, KV cache, decoding, serving tradeoffs, and backend performance constraints in production inference systems.
Experience with compiler technologies such as XLA, MLIR, LLVM, Pallas, or other kernel DSLs, including lowering, fusion, and backend code generation.
Knowledge of quantization methods such as INT8, FP8, mixed precision, or TPU-specific numeric formats, including accuracy and performance tradeoffs.
Bonus points if you have:
Contributed to vLLM, JAX/XLA, Pallas, PyTorch/XLA, compiler projects, or other open-source ML infrastructure.
Built TPU benchmarking infrastructure or automated performance regression detection for accelerator workloads.
Worked directly with Google TPU ecosystem stakeholders, accelerator platform teams, or early-access programs to ship backend, compiler, or inference performance improvements.
Location: This role is based in Singapore.
Compensation: Depending on background, skills, and experience, the expected annual salary range for this position is S$200,000 to S$400,000 annually + equity.
Visa sponsorship: We sponsor visas on a case-by-case basis.
Benefits: Inferact offers a generous benefits package, including medical, dental, and vision coverage.
Building the future of inference
Inferact is a startup founded by creators and core maintainers of vLLM, the most popular open-source LLM inference engine. Our mission is to grow vLLM as the world's AI inference engine and accelerate AI progress by making inference cheaper and faster.
Offices: San Francisco, CA, US
Building the future of inference
Inferact is a startup founded by creators and core maintainers of vLLM, the most popular open-source LLM inference engine. Our mission is to grow vLLM as the world's AI inference engine and accelerate AI progress by making inference cheaper and faster.
Offices: San Francisco, CA, US
