1 open role mentioning GPUDirect
Baseten
Baseten is seeking an engineer with strong machine learning experience and solid foundations in math and computer science to join our growing Post-Training team. This role is crucial for supporting the success of our customers' custom-trained models, which represent the majority of traffic on our platform. You will be responsible for building in-house tooling to efficiently train a wide spectrum of model architectures using various techniques at scale. This involves deep dives into technical topics and working across the full stack, from systems-level concepts like Kubernetes and storage to PyTorch distributed tensor computation and GPU kernels.
$200k - $275k