Founding GPU Compiler Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

SF Tensor · 1 week ago

Founding GPU Compiler Engineer

positionSan Francisco, CA
timeFull-time
remoteOnsite
senioritySenior Level
money$285K/yr - $315K/yr

SF Tensor is a company focused on rethinking the software and infrastructure stack for AI and high-performance computing. They are seeking a Founding GPU Compiler Engineer to build the core compilation infrastructure for their AI compiler, optimizing models from various frameworks into binaries for large-scale AI pre-training.

Artificial Intelligence (AI)Cloud ComputingMachine LearningSoftware

Responsibilities

Design and implement the main compilation pipeline, from StableHLO to executable GPU and host binaries
Build and extend MLIR dialects and passes to optimize AI workloads
Develop backend code generation for multiple targets (NVIDIA PTX/SASS, AMD GCN/RDNA, Trainium, TPU)
Implement classic compiler optimizations customized for large-scale training (fusion, tiling, memory planning, scheduling)
Build search-based compiler infrastructure to explore different optimization options
Create hybrid codegen paths for cases where direct MLIR lowering isn't practical
Set up testing, benchmarking, and performance regression systems
Work closely with ML researchers to understand workload characteristics and find optimization opportunities

Qualification

Compiler infrastructureGPU architectureC++RustMLIRCUDAPTX/SASSGCN/RDNAML compiler stacksProduction-grade compilerSoft skills

Required

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

Preferred

Background in distributed systems or multi-device compilation
Contributions to open-source compiler projects
Experience with autotuning or search-based optimization
Familiarity with large-scale training infrastructure
Experience with (Stable)HLO

Benefits

Bonus
Equity
Benefits
Relocation assistance

Company

SF Tensor

twittertwittertwitter
company-logo
The San Francisco Tensor Company is reinventing the software and infrastructure stack for modern AI and HPC.
dateFounded in 2025
location
San Francisco, California, USA
people-size2-10 employees
web-link
https://sf-tensor.com

Funding

Current Stage
Early Stage
Total Funding
$0.5M
Key Investors
Y Combinator
2025-10-01Pre Seed· $0.5M
Company data provided by crunchbase