GPU Compiler Engineer — End-to-End MLIR, RTL & Equity

SF Tensor; San Francisco (CA); On-site; USD 285,000 - 315,000; Full time

Benefits offered by this job

Relocation assistance

Equity and benefits

Job summary

SF Tensor in San Francisco is building the fastest GPU compiler in the world. We seek a Member of Technical Staff for GPU Compiler Engineering to own the end-to-end pipeline—from MLIR dialects to backend code generation and executable binaries.

You will extend MLIR, work on the LLVM backend, and drive performance across training and inference workloads for diverse hardware targets. Relocation assistance is available; base salary ranges from $285,000 to $315,000 with meaningful equity.

Qualifications

• Deep experience in compiler infrastructure (LLVM, MLIR or similar).
• Strong background in GPU architecture and low-level optimization (CUDA, ROCm or similar).
• Hands-on experience with PTX/SASS, GCN/RDNA assembly or other GPU ISAs.
• Familiar with ML compiler stacks (XLA, TVM, Triton, torch.compiler or similar).
• Solid systems programming skills in C++ and/or Rust.
• Proven track record of building production-grade compiler infrastructure.

Responsibilities

• Build and extend MLIR dialects and passes to optimize training and inference workloads.
• Work in our LLVM backend, below ptxas, on instruction selection, scheduling, register allocation and direct cubin emission.
• Expand our search-based compiler infrastructure with agent- and RL-driven program search and formal correctness proofs.
• Implement classic compiler optimizations tuned for large-scale training.
• Create hybrid codegen paths for cases where direct MLIR lowering isn't practical.
• Own testing, benchmarking and performance regression systems, including bit-identical hardware models.
• Collaborate with research and customer workloads to find optimization opportunities.

Skills

LLVM/MLIR

GPU architecture

C++ or Rust

GPU ISA

ML compiler stacks

Systems programming

Tools

LLVM backend

CUDA

ROCm

Job description

SF Tensor in San Francisco is building the fastest GPU compiler in the world. We seek a Member of Technical Staff for GPU Compiler Engineering to own the end-to-end pipeline—from MLIR dialects to backend code generation and executable binaries.

You will extend MLIR, work on the LLVM backend, and drive performance across training and inference workloads for diverse hardware targets. Relocation assistance is available; base salary ranges from $285,000 to $315,000 with meaningful equity.