GPU Kernel Engineer - High-Performance ML & Equity

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 environment, collaborating with research teams to productionize advances and contribute to open-source GPU libraries.