Remote Machine Learning Engineer

Summary

Join Helm.ai, a company building AI software for autonomous driving and robotics, as a Software Engineer to collaborate with research engineers and build infrastructure for research operations and automate the training and optimization of ML models.

Requirements

• 5+ years of experience in a directly related field
• Experience with tensorflow/pytorch, Jax, or related deep learning framework
• Experience building distributed, scalable software as a service
• Proficiency in Python
• Proven ability to thrive in a fast-paced environment
• Ability to communicate complex technical concepts to a variety of audiences
• Introspection, thoughtfulness, and detail-orientation

Responsibilities

• Write frameworks with flexible and modular components to quickly prototype trainings and implement new cutting-edge research ideas
• Build robust and fault-tolerant ML pipelines for running inference on large image and video datasets
• Leverage your knowledge of deep learning frameworks to optimize trainings in terms of GPU utilization, memory, speed, and accuracy
• Design and implement APIs to automate the training and validation of a wide range of ML models for specialized use cases

Preferred Qualifications

• Master or Ph.D. in Math, Computer Science or a related field
• Experience in learning and implementing techniques from research papers
• Computer vision experience

Benefits

• Competitive health insurance options
• 401K plan management
• Remote-friendly and flexible team culture
• Free lunch and fully-stocked kitchen in our South Bay office
• Additional perks: monthly wellness stipend, office set up allowance, company retreats, and more to come as we scale