AI-Driven Compiler Research Engineer (Relocation + Equity)

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

Benefits offered by this job

Relocation assistance

Meaningful equity

Job summary

SF Tensor is hiring a Member of Technical Staff for AI-Driven Compilation to own the search itself. You'll design RL systems that explore a massive program space and build agentic loops that use LLMs to reason about IR and transformations.

The role emphasizes rigorous experiments on real hardware with formal proofs of correctness, including tuning for latency on diverse targets like AMD, TPU, and Apple Silicon, with relocation assistance and a base salary of $275,000–$315,000 plus equity.

Qualifications

• Strong background in reinforcement learning with hands-on experience training agents.
• Experience building LLM agents, tool use and other agentic systems.
• Familiarity with GPU programming concepts and willingness to get down to the ISA.
• Proficient in PyTorch or JAX.
• Ability to design and run experiments that produce trustworthy results.

Responsibilities

• Design and implement RL systems that search over a massive program space: instruction selection, schedules, tile sizes, fusion strategies and phase ordering
• Build agentic compilation loops that use LLMs to reason about IR, propose transformations and learn from measured results
• Design the search and credit-assignment machinery around an exact reward built from measured latency on real hardware and a formal correctness proof
• Create representations and embeddings of compiler IR that hold up under learned optimization
• Build the training infrastructure for compiler optimization agents, including rollout throughput and distributed evaluation on real silicon
• Push transfer and cold-start performance so that the search works on unfamiliar targets and new ISAs from the first trial
• Close the loop with production workloads so that the compiler keeps improving from what customers actually run
• Work directly with our compiler and kernel engineers to land learned components in the shipping pipeline
• Run rigorous experiments, dig into the results, iterate on them and publish or open-source some of them

Skills

Reinforcement learning

LLM agents

GPU programming concepts

PyTorch

JAX

Experimentation design

Tools

XLA

TVM

Triton

MLIR

LLVM

Job description

SF Tensor is hiring a Member of Technical Staff for AI-Driven Compilation to own the search itself. You'll design RL systems that explore a massive program space and build agentic loops that use LLMs to reason about IR and transformations.

The role emphasizes rigorous experiments on real hardware with formal proofs of correctness, including tuning for latency on diverse targets like AMD, TPU, and Apple Silicon, with relocation assistance and a base salary of $275,000–$315,000 plus equity.