Software Engineer - Machine Learning and Infrastructure

About the role

We are looking for a talented ML Infrastructure engineer to join the Autonomy team and help build state-of-the-art model training, evaluation, and deployment pipelines to aid in the development of the perception modules of our self-driving vehicles. You will be responsible for providing solutions to empower machine learning development and optimize offboard training.

What you’ll be doing:

• Design and implement machine learning tooling and workflows for analysis & augmentation of real-world data
• Develop and optimize training pipelines in distributed environments
• Establish automated ETL pipelines
• Programmatically increase training efficiency of different neural network architectures
• Improve the developer experience and performance of our scalable ML platform
• Develop application/ML metrics to measure model, perception system, and overall self-driving performance; analyze for performance optimization opportunities

What you'll bring:

• BS or above in Computer Science/Engineering and 4+ years of industry experience in designing and implementing deep learning, ML, or data analytics infrastructure
• Experience working with production machine learning pipelines, from dataset collection and labeling to training and validation
• Knowledge in using common deep learning frameworks, e.g. Tensorflow, Pytorch, Caffe Skilled in C++ (11 or newer) and Python Demonstrated experience in profiling CPU/GPU code
• Experience with developing, running, and managing container orchestration systems like Kubernetes
• Ability to thrive in a fast-moving, collaborative, small team environment with little supervision

What makes you a strong fit:

• Experience working with data processing pipelines for training in the cloud (AWS, Azure, Google Cloud, etc.)Solid understanding of metrics, data analysis, and scientific evaluation
• Strong software engineering skills building well-designed, highly-maintained and high-reliability code used by other engineers
• Passion for sustainable energy and electric vehicle development

$160,000 - $220,000 a year

The US base salary range for this full-time position is $160,000 to $220,000 + equity + benefits + unlimited PTO

The range displayed on each job posting reflects the minimum and maximum target for new hire salaries for the position across all US locations. Within the range, individual pay is determined by work location, internal equity, and additional factors, including, but not limited to, job-related skills, experience, and relevant education or specialty training. Your recruiter can share more about the specific salary range during the hiring process.