Performance Kernel Engineer, GPU & ML Inference

INFERACT SINGAPORE PTE. LTD.; Penarth; On-site; GBP 99,000 - 198,000; Full time

Benefits offered by this job

Equity

Medical insurance

Dental insurance

Vision coverage

Job summary

Inferact is looking for a performance engineer to squeeze every FLOP from modern accelerators and to write kernels and low-level optimizations that power vLLM’s speed. Your code will run on hundreds of accelerator types, from NVIDIA GPUs to emerging silicon, and you’ll collaborate with hardware vendors to ensure maximum performance.

Fresh graduates are welcome to apply. Compensation includes equity and a comprehensive benefits package.

Qualifications

• Bachelor's degree in CS, engineering, or equivalent.
• Experience writing CUDA kernels or equivalent (CuTeDSL, Triton, TileLang, Pallas).
• Strong understanding of GPU architecture: memory hierarchy, warp scheduling, tiling, tensor cores.
• Proficiency in C++ and Python with high-performance coding.
• Experience with profiling tools (Nsight, rocprof) and optimization methodologies.
• Obsession with benchmarks and squeezing every percentage point of speedup.

Responsibilities

• Write kernels and low-level optimizations for vLLM across accelerator types.
• Collaborate with hardware teams to maximize performance on new chips.
• Tune and profile code to achieve high FLOPS and efficiency.
• Contribute to architecture-aware optimizations and open-source components.

Skills

CUDA kernels

C++

Python

GPU architecture

Performance profiling

Benchmark optimization

Education

Bachelor's degree in Computer Science or Engineering

Tools

Nsight

rocprof

CuTeDSL

Triton

TileLang

Pallas

LLVM

MLIR

XLA

Job description

Inferact is looking for a performance engineer to squeeze every FLOP from modern accelerators and to write kernels and low-level optimizations that power vLLM’s speed. Your code will run on hundreds of accelerator types, from NVIDIA GPUs to emerging silicon, and you’ll collaborate with hardware vendors to ensure maximum performance.

Fresh graduates are welcome to apply. Compensation includes equity and a comprehensive benefits package.