GPU Kernel Engineer
TypeSafe AI
3 days ago
Full-time
On-site
TypeSafe is a frontier model lab. We build reliable and general AI systems to power economically valuable automation. Our mission is to usher in a new era of Transformative Artificial Intelligence (TAI): technology with the power to drive a societal shift on the scale of the agricultural and industrial revolutions.
While others chase benchmarks and academic puzzles, we've been quietly rethinking the LLM stack from first principles — building a new kind of general frontier model designed for real-world reliability, decision-making, and autonomy in production.
We're a small, fast-moving team from OpenAI, Google Brain, and Meta/FAIR, backed by top-tier investors. Since mid-2024, we've been engineering the foundation for what comes after the current "state-of-the-art" — a model that actually gets things done.
About the role
We'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 TypeSafe
We'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
While others chase benchmarks and academic puzzles, we've been quietly rethinking the LLM stack from first principles — building a new kind of general frontier model designed for real-world reliability, decision-making, and autonomy in production.
We're a small, fast-moving team from OpenAI, Google Brain, and Meta/FAIR, backed by top-tier investors. Since mid-2024, we've been engineering the foundation for what comes after the current "state-of-the-art" — a model that actually gets things done.
About the role
We'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 TypeSafe
We'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