Machine Learning Engineer – Infrastructure and Automation | Helm.ai | Remote (Worldwide)

Machine Learning Engineer – Infrastructure and Automation | Helm.ai | Remote (Worldwide)

Remote Worldwide
Application ends: September 21, 2024
Apply Now

Job Description

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

We offer:

  • 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

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 of Helm.ai are considered the property of Helm.ai and are not subject to payment of agency fees.