Machine Learning Engineer

Summary

The job is for a Research Engineer at Helm.ai, working on AI software for autonomous driving and robotics. The position requires 5+ years of experience in a related field, proficiency in Python, and expertise with deep learning frameworks like TensorFlow/PyTorch or Jax.

Requirements

• 5+ years of experience in a directly related field
• Experience with tensorflow/pytorch, Jax, or related deep learning frameworks
• Proficiency in Python

Responsibilities

• Collaborate with research engineers to build infrastructure for research operations and automate the training and optimization of ML models for real-time inference in computer vision applications
• 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 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
• The opportunity to work on one of the most interesting, impactful problems of the decade