Founding Research Engineer, AI-Driven Compilation

Job description

ABOUT SF TENSOR At The San Francisco Tensor Company, we believe the future of AI and high-performance computing depends on rethinking the entire software and infrastructure stack. Today's developers face bottlenecks across hardware, cloud, and code optimization that slow progress before ideas can reach their full potential. Our mission is to remove those barriers and make compute faster, cheaper, and universally portable. We are building a Kernel Optimizer that automatically transforms code into its most efficient form, combined with Tensor Cloud for adaptive, cross-cloud compute and Emma Lang, a new programming language for high-performance, hardware-aware computation. Together, these technologies reinvent the foundations of AI and HPC. SF Tensor is proudly backed by Susa Ventures and Y Combinator, as well as a group of angels including Max Mullen and Paul Graham as well as founders and executives of NeuraLink, Notion and AMD. We are partnering with researchers, engineers, and organizations who share our belief that the next breakthroughs in AI require breakthroughs in compute. ABOUT THE ROLE We're building a next-generation AI compiler that uses machine learning to optimize machine learning. As a Founding Research Engineer focused on AI-driven compilation, you'll develop the agentic and reinforcement learning systems that guide our compiler's optimization decisions. This is a research-heavy role right at the intersection of compilers and ML. You'll design learned systems that can explore huge optimization spaces, discover new code transformations, and keep improving compilation quality over time. WHAT YOU'LL DO - Design and implement RL-based systems for compiler optimization (things like phase ordering, tile size selection, scheduling decisions, and fusion strategies) - Build agentic compilation systems that use LLMs to reason about code and apply transformations - Develop reward models and the training infrastructure for our compiler optimization agents - Create representations and embeddings of compiler IR that work well for learned optimization - Design feedback loops that let the system improve continuously from real production workloads - Work closely with compiler engineers to integrate these learned components into the full compilation pipeline - Run experiments, dig into the results, and iterate on what works - Publish and open-source research when it makes sense WHAT WE'RE LOOKING FOR - Strong background in reinforcement learning, with hands-on experience training RL agents on real problems - Experience building LLM agents, tool use, or other agentic systems - Good familiarity with GPU programming concepts - Solid proficiency in Python and PyTorch or JAX - Ability to design and run solid, rigorous experiments NICE TO HAVE - Experience with ML compiler stacks (XLA, TVM, Triton, MLIR) - Experience with RLHF, reward modeling, or preference learning - Background in combinatorial optimization or program synthesis - Publications or clear research contributions in RL, learned optimization, or ML for systems - Familiarity with compiler concepts (IR, optimization passes, code generation) - Familiarity with GPU performance optimization - Prior work on learned indexing, learned query optimization, or similar ML-for-systems projects WHY JOIN US Compiler optimization is one of the most exciting areas for applying learned systems. There are massive action spaces, clear reward signals, and huge real-world impact. You'll get the freedom to chase ambitious research ideas while building things that actually ship to production. We believe in the power of in-person collaboration to solve the hardest problems and foster a strong team culture. We offer relocation assistance and look forward to you joining us in our San Francisco office. The base salary range for this full-time position is $275,000 - $315,000 + bonus + equity + benefits.