Machine Learning Engineer, Infrastructure and Automation
Remote, United States
Helm.ai
Explore Helm.ai's AI software for high-end ADAS through Level 4 autonomous vehicles. Helm.ai offers a full suite of full-stack AI software, foundation models, and generative AI development and validation tools for production-bound projects.We help make autonomous technologies more efficient, safer, and accessible.
Helm.ai builds AI software for autonomous driving and robotics. Our Deep Teaching™ methodology is uniquely data and capital efficient, allowing us to surpass traditional approaches. Our unsupervised learning software can train neural networks without the need for human annotation or simulation and is hardware-agnostic. We work with some of the world's largest automotive manufacturers and we've raised over $100M from Honda, Goodyear Ventures, Mando, and others to help us scale.
Our team is made up of people with a diverse set of experiences in software and academia. We work together towards one common goal: to integrate the software you'll help us build into hundreds of millions of vehicles.
You will collaborate with research engineers to build infrastructure to perform research operations and automate the training and optimization of ML models for real-time inference in computer vision applications. Specifically, you will:
- 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
You have:
- 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
The following are a plus but not required:
- Master or Ph.D. in Math, Computer Science or a related field
- Experience in learning and implementing techniques from research papers
- Computer vision experience
- 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
* Salary range is an estimate based on our AI, ML, Data Science Salary Index 💰
Tags: APIs Autonomous Driving Computer Science Computer Vision Deep Learning GPU Helm JAX Machine Learning Mathematics ML models Pipelines Python PyTorch Research Robotics Teaching TensorFlow Unsupervised Learning
Perks/benefits: 401(k) matching Insurance Snacks / Drinks Team events Wellness
More jobs like this
Explore more career opportunities
Find even more open roles below ordered by popularity of job title or skills/products/technologies used.