Machine Learning Engineer, Infrastructure and Automation

Helm.ai; United States; Remote; USD 80,000 - 150,000; Full time

Job summary

An innovative company is seeking a Machine Learning Engineer to enhance autonomous technologies. You will work on cutting-edge AI software for autonomous driving and robotics, collaborating with research engineers to build infrastructure for real-time ML model training and optimization. This role offers the chance to tackle impactful challenges while working in a remote-friendly environment with a flexible team culture. Join a diverse team that values creativity and technical excellence, and contribute to shaping the future of autonomous vehicles.

Benefits

Competitive health insurance options

Remote-friendly culture

Free lunch

Monthly wellness stipend

Office setup allowance

Company retreats

Qualifications

• 5+ years of experience in a related field.
• Experience with deep learning frameworks like TensorFlow or PyTorch.
• Ability to thrive in a fast-paced environment.

Responsibilities

• Collaborate with research engineers to build infrastructure for ML operations.
• Write frameworks to prototype trainings and implement research ideas.
• Design APIs to automate training and validation of ML models.

Skills

Python

TensorFlow

PyTorch

JAX

Deep Learning

Communication Skills

Education

Bachelor's Degree in Computer Science

Master's or Ph.D. in Math or Computer Science

Tools

ML Pipelines

APIs

Job description

Machine Learning Engineer, Infrastructure and Automation

Remote, United States

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

We offer:

• Competitive health insurance options
• 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

Helm.ai is proud to be an equal opportunity employer building a diverse and inclusive workforce. All applicants will be considered for employment without attention to race, color, religion, sex, sexual orientation, gender identity, national origin, veteran or disability status.

Any unsolicited resumes/candidate profiles submitted through our website or to personal email accounts of employees ofHelm.ai are considered the property ofHelm.ai and are not subject to payment of agency fees.

Apply for this job

*

indicates a required field

First Name *

Last Name *

Email *

Phone *

Resume/CV *

Enter manually

Accepted file types: pdf, doc, docx, txt, rtf

Enter manually

Accepted file types: pdf, doc, docx, txt, rtf

In lieu of a cover letter, feel free to share what you find most attractive about working at a startup in the autonomous vehicle space.

Where are you located? *

Would you require us to sponsor a work visa for your location? * Select...

Were you referred by a Helm employee? If yes, please list their name.