Machine Learning Systems Engineer

We are looking for a Machine Learning Systems Engineer to join our ML Acceleration team. In this role, you will be responsible for the core systems that enable our researchers to train frontier models at scale, focusing obsessively on speed, cost, reliability, and throughput.

Requirements

• Bachelor's, Master's degree, or PhD in Computer Science, Computer Engineering, or a related technical discipline.
• Strong proficiency in Python.
• Extensive hands-on experience with PyTorch.
• Experience optimizing machine learning model execution during training and inference, alongside a strong understanding of fundamental machine learning concepts, architectures, and processes.
• Exceptional analytical and problem-solving skills, with a bias for action and a data-driven approach to technical challenges.

Benefits

• Medical
• Dental
• Vision
• 401k with a company match
• Health saving accounts
• Life insurance
• Pet insurance