---
title: 'Machine Learning Engineer at UnitX'
canonical: 'https://feeny.ai/job/machine-learning-engineer-unitx-hq-t9aayydq3v8s'
type: 'job'
last_seen: '2026-09-10'
---

# Machine Learning Engineer at UnitX

- **Company:** UnitX
- **Location:** HQ
- **Employment:** full-time
- **Work type:** onsite
- **Posted:** 2026-06-30
- **Last confirmed live:** 2026-09-10
- **Apply:** https://jobs.ashbyhq.com/unitxlabs/d75b1e49-8513-4c21-a879-10678d64e056

## Job description

Job Title: Machine Learning Engineer

About Us: UnitX builds the world's 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 1,000+ mission-critical systems across 190+ of the world's leading manufacturers' production lines. Every year, $15B worth of products go through UnitX's 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 You'll 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: Bachelor's 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: $174,304 – $200,000 per year. (Final compensation is based on experience).
- Full Medical, Dental, Vision, 401k
- Unlimited PTO
- Daily meals provided

How to apply Please send your resume and a cover letter to james@unitxlabs.com and ensure you include Ref#MM0426 in the subject line.

## About UnitX

## Company Overview
- **One-liner**: UnitX provides AI-powered visual inspection systems that automate inline defect detection for automotive, EV battery, and electronics manufacturing.
- **Entity Type**: Private (Series B, $106M total funding)
- **Headquarters**: Santa Clara, California, United States
- **Founded**: 2018
- **Founders**: Keven Wang (CEO), Weixiong Zheng (CTO), Adam Yang (COO)

## Core Business
- **Primary Industry**: Industrial Automation / AI-powered manufacturing inspection
- **Target Customers**: B2B, Enterprise — automotive OEMs and Tier 1 suppliers, EV battery manufacturers, electronics manufacturers, PCBA, connector, and semiconductor producers
- **Mission**: To elevate 20% of the world's GDP through intelligent automation, building the world’s best team in robotics

## Products & Services
- **CorteX**: AI training and inference system — rapidly trains, deploys, and manages yield-optimized AI models to detect and classify variable defects with high-speed, inline decisions.
- **OptiX**: Fully software-defined imaging system that captures 2D and 2.5D features inline for edge inference.
- **FleX-Gen**: Turnkey inline inspection equipment for high-speed production lines.
- **YieldX**: Analytics platform delivering data reports for AI-powered manufacturing optimization.

## Market Standing
- **Valuation**: Not publicly available
- **Total Funding**: $106M across four rounds (Seed: $2M, Series A: $12M, Series B (June 2023): $41M, Series B extension (Nov 2024): $46M)
- **Lead Investors**: UP Partners (led Series A and most recent Series B), SE Ventures (led 2023 venture round)
- **Key Metric**: Systems currently inspect more than **$6.1 billion worth of products annually**; deployed on **100+ production lines**.
- **Customer Base**: Served 190+ customers, including 10 of the world’s top 50 automotive manufacturers, 10 of the top 30 Li Battery makers, 9 of the top 100 electronics manufacturers, and 2 of the top 4 EV manufacturers.
- **Growth Signals**: +19.6% headcount growth year-over-year; +60% job posting growth quarterly; global operations across 8 countries (US, Mexico, China, Vietnam, Thailand, Kenya, Japan, Hong Kong).

## Competitive Advantages
- **Superhuman defect detection**: AI models that detect and classify complex defects with accuracy exceeding human inspection, regardless of part, material, or position.
- **Fastest deployment**: Small-sample AI that integrates with existing imaging systems; software-defined lighting enables rapid adaptation to new environments — claim of fastest-to-deploy inline inspection in the industry.
- **Proven ROI**: Customers see a typical <12 month ROI with average $1.3M returned per production line and 3% scrap reduction.
- **Track record**: 5.7 million+ continuous operating hours across deployed systems.

## Strategic Focus
- Deepening penetration in high-growth verticals: EV battery manufacturing, automotive components, and consumer electronics.
- Enabling global scale with a hardware + software + AI platform model (imaging, inference, and analytics).
- Building synthetic defect generation capabilities to accelerate model training and improve yield outcomes for customers.

## Why Work Here
- **Culture & Team**: Founded by engineers from Stanford, MIT, and Google; described internally as a robotics company with values of reliability, precision, transparency, and ambition.
- **Growth Trajectory**: 78 employees with rapid expansion (+19.6% YoY); 8 active job openings including Embodied AI Engineer, Machine Learning Engineer, Simulation Engineer, and Recruiter roles.
- **Work Environment**: Santa Clara HQ with a global, cross-functional team (Sales 25%, Technical 24%, Operations 5%). Talent sourced from top companies (Keyence, Amazon, Stanford, Palo Alto Networks, Verkada) — suggests high-performance culture.
- **Perks & Mission**: Direct impact on worldwide manufacturing quality; opportunity to work on full-stack robotics combining deep learning, imaging, and hardware deployed at scale.

## Sources
1. [unitxlabs.com](https://www.unitxlabs.com/about-us/)
2. [unitxlabs.com](https://www.unitxlabs.com/)
3. [linkedin.com](https://www.linkedin.com/company/unitxlabs)
4. [craft.co](https://craft.co/unitxlabs)
5. [jobs.ashbyhq.com](https://jobs.ashbyhq.com/unitxlabs/42762a56-2207-4e72-bf90-f5321683c638)

## Other roles at UnitX

- [Solutions Manager](https://feeny.ai/job/solutions-manager-unitx-hq-j2cw0rzdkv6z) — HQ
- [Director, Business Development & Strategic Partnerships](https://feeny.ai/job/director-business-development-strategic-partnerships-unitx-hq-ryfgm00rffxb) — HQ
- [Product Manager, Vision & Robotics](https://feeny.ai/job/product-manager-vision-robotics-unitx-hq-0smxt25hnwcq) — HQ
- [Technical Recruiter](https://feeny.ai/job/technical-recruiter-unitx-hq-bjsk2kkq96f7) — HQ
- [Field Deployment Manager](https://feeny.ai/job/field-deployment-manager-unitx-hq-gm6y2rbw412h) — HQ
- [Account Executive](https://feeny.ai/job/account-executive-unitx-remote-ptaqc8tp3azt)
- [Product Operations Engineer](https://feeny.ai/job/product-operations-engineer-unitx-hq-zstafp45zxfb) — HQ
- [Business Development Representative](https://feeny.ai/job/business-development-representative-unitx-remote-y3crmw7zh1a8)
- [Machine Learning Engineer](https://feeny.ai/job/machine-learning-engineer-unitx-hq-8cjdawnk76qr) — HQ
- [Embodied AI Engineer](https://feeny.ai/job/embodied-ai-engineer-unitx-hq-m18ckepm5k35) — HQ