ML Systems Engineer: Inference & GPU-Driven Distributed Workloads

Bake AI in San Mateo, California, is hiring a Member of Technical Staff to build ML systems that enable sustained AI research. You will own performance-critical components across model serving, GPU execution, and distributed workloads, and you will push beyond a simple inference engine to improve correctness and efficiency. You will work with researchers and infrastructure engineers, design robust systems, measure performance, and drive implementation from architecture through production, while