GPU Kernel Engineer

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