Founding Kernel Engineer
San Francisco, CA at a glance
- Rent
- #2 of 51$2,680/mo+46% vs US avg
- Weather
- #17 of 51295 mild days0 hot · 0 cold
- Income tax
- #1 of 5113.3% top rateCalifornia
What you need
- Hands-on experience developing GPU kernels
- CUDA, Triton, or GPU instruction sets experience
- MLIR/LLVM or PyTorch backends knowledge
- Hardware performance analysis and benchmarking skills
- Rust or open-source compiler contributions
What you'll do
- Research and build faster kernels for AI workloads
- Investigate performance bottlenecks with the compiler team
- Take optimizations through correctness checks and benchmarking
- Execute optimizations on real hardware
- Develop agent-driven systems for kernel research
Qualifications
- Hands-on experience developing GPU kernels.
Nice to haves
- Prior work on kernel autoresearch frameworks is a very strong positive.
- Experience with ML compilers is a strong positive, but not required.
- Hardware performance analysis, memory hierarchies, parallel execution, and benchmarking.
- CUDA, Triton, GPU instruction sets, or compiler systems such as MLIR/LLVM and PyTorch backends.
- Rust, open-source compiler contributions, or work on other accelerators.
No degree requirement. Researchers, recent PhDs, and candidates with substantial relevant projects are welcome.
Company
Luminal (YC S25) builds an AI compiler and serving stack that optimizes models for faster, more efficient execution across accelerators. We're a small team working in person in San Francisco.
Role
Research and build faster kernels for AI workloads. You'll work closely with the compiler team, investigate performance bottlenecks, and take optimizations through correctness checks, benchmarking, and execution on real hardware.
Our focus includes:
- Agent-driven systems that write, test, and improve kernels.
- Automated kernel research and search-based performance optimization.
- More efficient execution for inference and LLM training.
Candidates who require a US visa are welcome to apply.
Optimize your resume for this job
Get a match score and the keywords you're missing
About Luminal
Luminal provides a machine learning compiler and serverless cloud platform that automates PyTorch model optimization and deployment.
Similar roles
