GPU Kernel Engineer

TypeSafe AIGPU Kernel Engineer LocationSan Francisco OfficeEmployment TypeFull timeLocation TypeOn-siteDepartmentMember of Technical StaffCompensationBase salary: $180K – $280K • Offers EquityAbout TypeSafeTypeSafe 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 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 inferenceProfile end-to-end performance and eliminate bottlenecks across the stackPartner with research and platform engineers to squeeze maximum throughput and minimum latency out of our hardwareWe are looking for people whoHave deep CUDA / GPU kernel expertise and a track record of real performance winsHave built and optimized inference / training kernelsHave hands-on LLM training experience (real, not at a hobbyist level)Reason from first principles about performance, memory, and parallelismAre responsible, ownership-inclined team players who are mission aligned and excited to go all-inLife 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 provideBase salary of $180k–280k plus equity, based on leveling100% covered health insuranceDaily lunch and dinnerVisa sponsorships401K plansApply for this Job