Machine Learning Engineer

Berkeley, California

Ambi Robotics

A robotics company that creates advanced AI-powered robotic systems designed to scale and handle increasing demand in the supply chain.

View company page

We are a well-funded Series A startup on a mission to make e-commerce logistics more human-friendly and efficient with AI-powered robotics.
Founded by leading roboticists, we are growing our team of dynamic and compassionate tech enthusiasts to deploy robots at scale. Join us in building the most dexterous robots on the planet!
We seek a Machine Learning Engineer to build software infrastructure for training, deploying, versioning, monitoring, and updating a suite of deep learning models in production. In this role, you will work closely with the CTO, VP of Software, Product Lead, and Head of ML/CV.

In this role, you will architect, develop, and maintain:

  • Machine learning infrastructure for prototyping, training and deploying different deep learning models.
  • Continuous learning pipelines for dynamically curating new training datasets of out-of-distribution inputs and failure modes that interact with data labeling infrastructure (in-house and third-party) in a turn-key fashion.
  • CI/CD pipelines for deploying new models with online experiments informed by statistical methods such as A/B testing.
  • Model registry for a variety of ML models, cfgs, on-prem and cloud datasets.
  • Instrumentation and tooling for monitoring/debugging live models.

You're a good fit if you have:

  • Eagerness to learn, solve challenging problems, and lead with curiosity.
  • A desire for career growth and ownership of your work.
  • Excitement about amplifying human potential with state-of-the-art robotics.
  • 2+ years of experience in a professional ML SWE role, working with ML teams in a production environment.
  • Proficiency with Python, PyTorch, and OOP design/implementatoin/testing.
  • Expertise with general SWE practices/workflow including design documentation, code review, API design/implementation, containerization, and cloud computing.
  • Familiarity with basic machine learning theory (e.g. supervised learning).
  • Experience with CI/CD pipline design and implementation, including cloud orchestration.
  • Comfortable working in a Linux dev environment.
  • (BONUS) Experience with AWS, C++/Java, Node, ML Ops, sys admin practices,

In addition to competitive compensation, we offer benefits such as:

  • Health, dental, and vision insurance
  • 401k with 5% matching by Ambi
  • Unlimited PTO
  • Access to a full gym at HQ
  • Free parking on site at HQ
  • Free lunches 2x per week
  • Equity ownership
Ambi Robotics is an equal opportunity employer. We celebrate diversity and are committed to creating an inclusive environment for all employees. We encourage and embrace applicants of all backgrounds to apply! 
DisclaimersAmbi Robotics provides equal employment opportunities to all employees and applicants for employment and prohibits discrimination and harassment of any type without regard to race, color, religion, age, sex, national origin, disability status, genetics, protected veteran status, sexual orientation, gender identity or expression, or any other characteristic protected by federal, state or local laws. This policy applies to all terms and conditions of employment, including recruiting, hiring, placement, promotion, termination, layoff, recall, transfer, leaves of absence, compensation and training.
Ambi Robotics will comply with federal and state mandates regarding COVID-19 vaccinations. We will require employees coming on-site to be vaccinated or in compliance with weekly testing requirements. Please note this job description is not designed to cover or contain a comprehensive listing of activities, duties or responsibilities that are required of the employee for this job. Duties, responsibilities and activities may change at any time with or without notice.

* Salary range is an estimate based on our AI, ML, Data Science Salary Index 💰

Tags: A/B testing APIs AWS CI/CD Deep Learning E-commerce Linux Machine Learning ML infrastructure ML models OOP Pipelines Prototyping Python PyTorch Robotics Statistics Testing

Perks/benefits: Career development Competitive pay Equity Fitness / gym Health care Insurance Lunch / meals Salary bonus Startup environment Team events Unlimited paid time off

Region: North America
Country: United States
Job stats:  12  4  1

More jobs like this

Explore more AI, ML, Data Science career opportunities

Find even more open roles in Artificial Intelligence (AI), Machine Learning (ML), Natural Language Processing (NLP), Computer Vision (CV), Data Engineering, Data Analytics, Big Data, and Data Science in general - ordered by popularity of job title or skills, toolset and products used - below.