Baseten is seeking a GPU Kernel Engineer to design and optimize high-performance GPU kernels for AI workloads. You will work on low-level optimization, leveraging CUDA C++ and tensor cores to accelerate core ML operations, with exposure to state-of-the-art inference systems.
The role offers growth potential in a fast-paced setting, collaborating with research teams to productionize advances and contribute to open-source GPU libraries.