Machine Learning Engineer - Infrastructure and Automation

August 13

Apply Now

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. β€’ 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

Requirements

β€’ 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

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

Apply Now

Similar Jobs

Built byΒ Lior Neu-ner. I'd love to hear your feedback β€” Get in touch via DM or lior@remoterocketship.com