Job details

Machine Learning Engineer - Infrastructure and Automation

Helm.ai is seeking a Senior Machine Learning Engineer to work on infrastructure and automation for autonomous technologies. The ideal candidate will have experience in creating robust ML pipelines and optimizing trainings for real-time inference in computer vision applications.

Skills

Experience with tensorflow/pytorch, Jax, or related deep learning framework
Proficiency in Python
Ability to thrive in a fast-paced environment
Excellent communication skills

Responsibilities

Write frameworks with flexible and modular components for prototype trainings and implement cutting-edge research ideas
Build fault-tolerant ML pipelines for large image and video datasets
Optimize trainings in terms of GPU utilization, memory, speed, and accuracy
Design and implement APIs to automate training and validation of ML models

Education

Bachelor's degree in a related field
Master or Ph.D. in Math, Computer Science or related field (preferred)

Benefits

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

To read the complete job description, please click on the ‘Apply’ button

Helm.ai Glassdoor Company Review

No rating