Principal Engineer, ML Training Platform

Summary

Join our team as a proactive, collaborative Principal Engineer for our ML Training Platform team. In this role, you will lead a driven, high-visibility team to deliver on key engineering objectives and be the accountable owner of the ML training platform.

Requirements

• 6+ years of Python software development experience
• Hands-on experience with popular ML frameworks (PyTorch or TensorFlow)
• Hands-on experience with scaling ML systems
• Practical experience with large scale AWS cloud infrastructure utilizing Kubernetes
• Strong problem solving skills and ability to evaluate challenges with an objective, data-driven approach
• Excellent programming and software design skills, including debugging, performance analysis, and test design
• Proven track record of operating highly-available systems at scale
• Strong collaboration and mentorship skills

Responsibilities

• Design, build and maintain scalable ML data processing, model training solutions in the AWS cloud infrastructure environment utilizing Kubernetes
• Perform training and model performance optimization with various GPUs to improve model training speed and efficiency
• Leverage Pytorch and Ray deep learning frameworks to operate highly-available systems at scale
• Drive the execution of technical programs and ensure milestone delivery
• Actively manage and mitigate technical risks

Preferred Qualifications

• Hands-on experience with Ray in large scale environments
• Experience with ML data processing for large-scale deep learning training
• Experience refactoring ML code written by ML engineers

Benefits

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