← All roles

Member of Technical Staff, AI-Driven Compilation

SF Tensor· SF Tensor / AI-Driven Compilation
Apply on company site
Hybrid (San Francisco)$275k - $315kFull-timePosted 1mo ago

About the role

Role focused on building the fastest GPU compiler using agentic and RL-based exploration to optimize programs across vendors and ISAs, with production-focused integration and verification.

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

Qualifications

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

Skills

Required:Reinforcement learningLLM agents / agentic systemsGPU programming conceptsPyTorch or JAX

Benefits

  • Meaningful equity and benefits
  • relocation assistance
Last updated Oct 7, 2026. Interested? Apply on the company site.
SF Tensor logo

About SF Tensor

Infrastructure for AI labs to focus on research.

AI researchers should be pushing the boundaries of what's possible with new architectures and training methods. Instead, they waste weeks configuring cloud infrastructure, debugging distributed systems, and optimizing their GPU code. We know because we lived it: While training our own models across thousands of GPUs earlier this year, we spent more time fighting our infrastructure than doing actual research. That why we're building two things. First, Elastic Cloud: a managed platform that automatically finds the cheapest GPUs across all providers, handles spot instance preemption, and cuts compute costs by up to 80%. Second, automatic kernel optimization that makes training code run faster by modeling hardware topology, often beating hand-tuned implementations. The problem is that getting high performance across different hardware is genuinely hard. NVIDIA's CUDA moat exists because writing fast kernels requires deep expertise. Most teams either accept vendor lock-in or hire expensive kernel engineers. Our goal is to break the CUDA moat. The compute bottleneck is the biggest constraint on AI progress. NVIDIA can't manufacture enough GPUs, and their monopoly keeps prices astronomical. Meanwhile, AMD, Google, and Amazon are shipping capable alternative hardware that nobody uses because the software is too hard. We're breaking that moat. If we succeed, anyone will be able to train state-of-the-art models without thinking past their PyTorch code.

machine-learningopen-sourcecloud-computingai

More open roles at SF Tensor