Impress employers and recruiters.
Choose from hundreds of resume examples.
Impress employers and recruiters.
Choose from hundreds of resume examples.
Target Resume Now
About the Role
We are building the fastest GPU compiler in the world, which proves correctness at the end of transformations to allow for a wider search space. We are hiring a Member of Technical Staff for AI-Driven Compilation to own the search process, building agentic and RL systems to optimize program transformations.
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
Requirements
- 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
- Proficiency in PyTorch or JAX
- Ability to design and run experiments that produce trustworthy results
Skills
- Reinforcement Learning
- LLM Agents
- GPU Programming
- PyTorch
- JAX
- Compiler IR
- Experiment Design
Location
- San Francisco
Work Type
- Full-time
- Onsite
Experience Level
- Member of Technical Staff
Salary/Compensations
- $275,000-$315,000
Benefits
- Meaningful equity
- Relocation assistance
About the Company
- SF Tensor is building the future of high-performance compute for AI by rethinking and rebuilding the compute stack from hardware to cloud.
- We are backed by Susa Ventures, Y Combinator, and other notable investors and industry leaders.
- We are a small team that has achieved significant milestones, including pre-training foundation models on 4,000 AMD GPUs and designing large-scale compute clusters and supercomputers.