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
Please send your resume and a cover letter to [email protected] and ensure you include Ref#MM0426 in the subject line.
Skills Required
- Bachelor's degree in Computer Science, Mathematics, Statistics, or a related discipline (or foreign equivalent).
- 3+ years of experience in the job offered or a related occupation.
- At least 3 years hands-on experience with Deep Learning & Neural Networks.
- At least 3 years hands-on experience in Machine Learning Pipeline Development.
- At least 3 years hands-on experience in Sensor Fusion & Signal Processing.
- At least 3 years hands-on experience in Computer Vision & Real-Time Inference Optimization.
- At least 3 years hands-on experience in Statistical Modeling & A/B Testing.
What We Do
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.
