Kernel Engineer (Internship and Full-time)
Tilde Research San Francisco
What they're looking for
- Design, develop, and tune custom GPU kernels for core model operations
- Work with ML engineers to prototype and scale novel model architectures
- Contribute to system-wide efforts to improve efficiency and throughput, beyond just kernel-level optimizations
- Have experience in deep learning or related research areas
- Strong open source contributions
The full posting
Published by Tilde Research. Original posting
Tilde Research is a moonshot AI lab advancing mechanistic interpretability, new architectures, and pretraining science. We build foundational understanding of models to advance the frontier of intelligence.
About the role:
As a Kernel Engineer at Tilde, you'll design, implement, and optimize high-performance GPU kernels that are critical to scaling our training and inference workloads. Your work will enable faster iteration cycles, higher throughput, and lower latency. You'll work closely with ML researchers and engineers to co-design models and infrastructure that are deeply performance-aware, and help push the limits of what current hardware can support.
What you might work on:
- Design, develop, and tune custom GPU kernels for core model operations
- Work with ML engineers to prototype and scale novel model architectures
- Contribute to system-wide efforts to improve efficiency and throughput, beyond just kernel-level optimizations
You're a good fit if you:
- Have experience in deep learning or related research areas
- Have demonstrated exceptional capability in working on ML kernels. This can include:
- Strong open source contributions
- Thoughtful technical blog posts/work logs
- Previous experience working on hardware-aligned algorithms
- Deep familiarity with PyTorch, Triton/TK/TileLang (>1 of), basic familiarity with CUDA, and knowledge of GPU architecture.
- Communicate clearly and effectively, both verbally and in writing
- Strong algorithmic thinker
- Are able to learn quickly
More internships at Tilde Research
All Tilde Research internships
Don't apply to Tilde Research internships one at a time.
OfferAIO fills out the forms for you, tells you the day a new one opens, and keeps every interview and contact in one place. Your first ten applications are free.
Start free