GPU Kernel Engineer — Kernel Optimization & Compiler Innovation

SF Tensor is building the fastest GPU compiler and a versatile kernel optimizer. We seek a Member of Technical Staff for GPU Kernel Engineering to push the envelope on what hardware can do before any search runs. You will write and hand-optimize kernels that execute during pre-training, post-training and inference across NVIDIA, AMD, TPU and Trainium. You will convert kernel learnings into compiler-search structures, with access to an end-to-end stack down to the ISA.