About the roleWe're looking for a GPU kernel engineer with deep, low-level CUDA expertise to make our training and inference faster and more efficient. You'll write and optimize custom kernels, profile and eliminate bottlenecks, and work close to the metal across our model stack.
Responsibilities include:
- Write, optimize, and maintain high-performance GPU kernels (e.g., in CUDA / CuTe DSL) for training and inference
- Profile end-to-end performance and eliminate bottlenecks across the stack
- Partner with research and platform engineers to squeeze maximum throughput and minimum latency out of our hardware
We are looking for people who- Have deep CUDA / GPU kernel expertise and a track record of real performance wins
- Have built and optimized inference / training kernels
- Have hands-on LLM training experience (real, not at a hobbyist level)
- Reason from first principles about performance, memory, and parallelism
- Are responsible, ownership-inclined team players who are mission aligned and excited to go all-in
Life at TypeSafeWe're a small, flat, close-knit team dedicated to real-world impact preparing the world for Transformative AI. Our team works fully in-person in our San Francisco office near Embarcadero station. We love what we do and care about our work a lot.
We strive for excellence and craftsmanship and won't stop until we get there. When the team wins, we all win, and we enjoy collaborating and inspiring each other to grow as a team and as individuals.
We also value emotional honesty, kindness, and bringing your whole self to work. We build machines; we don't try to be machines.
We want you to be able to do the most impactful work of your career at TypeSafe and help define our future as a company.
We provide- Base salary of $180k-280k plus equity, based on leveling
- 100% covered health insurance
- Daily lunch and dinner
- Visa sponsorships
- 401K plans
