Milpitas, California, United States · On-site
$170k–$200k/yr
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,…
Skills: Deep Learning, Neural Networks, Machine Learning Pipeline Development, Sensor Fusion, Signal Processing
Sign up with Clera and we'll reach out the moment a role actually fits you — no more spraying applications into the void.
Design and optimize deep learning models for defect detection and segmentation using high-resolution images and 3D sensors. Develop scalable ML pipelines for edge and cloud deployment while collaborating with cross-functional teams to integrate solutions into industrial workflows.
Requires a Bachelor's degree in Computer Science, Mathematics, Statistics, or a related field. Must have at least 3 years of hands-on experience in deep learning, ML pipelines, sensor fusion, and computer vision optimization.
Machine Learning Engineer
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.
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.
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
Salary Range: $174,304 – $200,000 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 [email protected] and ensure you include Ref#MM0426 in the subject line.
Innovative deep learning inspection system for automated manufacturing.
UnitX was built in 2018 by a team of engineers from Stanford, MIT, and Google with deep respect for the manufacturing industry, UnitX is a robotics company automating repetitive visual tasks in manufacturing industries. We have made this physically demanding and unsafe job an innovative, precise, and easy-to-deploy process all while using deep learning and robotics. UnitX has deployed our product on production lines throughout the country where we have increased the defect inspection rate while simultaneously amplifying the cycle time far above that of a human inspector. We are building the world’s best team in robotics. Join us in building out an industry where we automate repetitive tasks and maximize human intellect! Visit to view our open positions.
Offices: Santa Clara, CA, US
Innovative deep learning inspection system for automated manufacturing.
UnitX was built in 2018 by a team of engineers from Stanford, MIT, and Google with deep respect for the manufacturing industry, UnitX is a robotics company automating repetitive visual tasks in manufacturing industries. We have made this physically demanding and unsafe job an innovative, precise, and easy-to-deploy process all while using deep learning and robotics. UnitX has deployed our product on production lines throughout the country where we have increased the defect inspection rate while simultaneously amplifying the cycle time far above that of a human inspector. We are building the world’s best team in robotics. Join us in building out an industry where we automate repetitive tasks and maximize human intellect! Visit to view our open positions.
Offices: Santa Clara, CA, US
