Machine Learning Systems Engineer

We are looking for a Machine Learning Systems Engineer to join our ML Acceleration team. The role involves optimizing machine learning systems, developing high-performance GPU kernels, and optimizing distributed training pipelines. You will work on scaling large-scale distributed model training and reducing time-to-convergence for our next generation of models.

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