Machine Learning Engineer - Infrastructure and Automation
August 13
🏡 Remote – Anywhere in California
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
