Machine Learning Engineer

Job Title:

Machine Learning Engineer

About Us:

UnitX builds the worlds leading physical AI systems to automate repetitive visual tasks in factories. UnitX is a fast-moving startup with a team from Stanford MIT Google and beyond. Since inception UnitX has deployed 1000 mission-critical systems across 190 of the worlds leading manufacturers production lines. Every year $15B worth of products go through UnitXs AI inspection system to ensure quality.

Join us for the rare opportunity to work on computer-vision-driven products that go beyond the state of the art and that improve global manufacturing efficiency.

What Youll Do:

• Develop & Optimize Machine Learning Models: Design train and optimize deep learning models for defect detection and segmentation using high-resolution images and 3D sensor inputs. You will implement advanced computer vision techniques (including Stable Diffusion Segment Anything Model (SAM) and transformer-based architectures) to ensure pixel-level precision and real-time inference.

• Build Automated Pipelines: Develop and maintain automated pipelines for data preprocessing augmentation and model training to continuously optimize deployed solutions.

• Deploy & Maintain Production-Ready Software: Develop scalable ML pipelines for deployment on both edge devices and cloud-based systems. You will optimize model inference for real-time decision-making ensuring processing times remain under 20 milliseconds.

• Collaborate Cross-Functionally: Work closely with software engineers robotics specialists and hardware teams to seamlessly integrate ML solutions into fully automated industrial workflows.

• Monitor & Ensure Reliability: Implement automated monitoring and alerting systems to track model drift identify failures and ensure 24/7 fault tolerance in a manufacturing environment.

• Develop Evaluation & Interpretability Tools: Create frameworks to assess model accuracy precision recall and efficiency. You will design visualization dashboards conduct A/B testing and leverage explainable AI (XAI) to improve transparency in quality assurance.

• Drive R&D & Cutting-Edge Solutions: Investigate emerging AI/CV techniques including self-supervised learning domain adaptation and multimodal data fusion. You will develop novel architectures incorporating reinforcement learning and Generative AI and contribute to academic publications patents and open-source projects.

Who You Are:

• Education: Bachelors degree (or foreign equivalent) in Computer Science Mathematics Statistics or a closely related discipline.

• Experience: 3 years of experience in the job offered or a related occupation.

• Technical Expertise: Must possess at least 3 years of hands-on experience in the following areas:

  • Deep Learning & Neural Networks

  • Machine Learning Pipeline Development

  • Sensor Fusion & Signal Processing

  • Computer Vision & Real-Time Inference Optimization

  • Statistical Modeling & A/B Testing

Compensation & Benefits

• Salary Range: $174304 $200000 per year. (Final compensation is based on experience).

• Full Medical Dental Vision 401k

• Unlimited PTO

• Daily meals provided

Please send your resume and a cover letter to and ensure you include Ref#MM0426 in the subject line.

Required Experience:

IC