UnitX

Job Title:Machine Learning EngineerAbout 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:<ul><li>Develop &amp; 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.</li><li>Build Automated Pipelines: Develop and maintain automated pipelines for data preprocessing, augmentation, and model training to continuously optimize deployed solutions.</li><li>Deploy &amp; 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.</li><li>Collaborate Cross-Functionally: Work closely with software engineers, robotics specialists, and hardware teams to seamlessly integrate ML solutions into fully automated industrial workflows.</li><li>Monitor &amp; Ensure Reliability: Implement automated monitoring and alerting systems to track model drift, identify failures, and ensure 24/7 fault tolerance in a manufacturing environment.</li><li>Develop Evaluation &amp; 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.</li><li>Drive R&amp;D &amp; 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.</li></ul>Who You Are:<ul><li>Education: Bachelor's degree (or foreign equivalent) in Computer Science, Mathematics, Statistics, or a closely related discipline.</li><li>Experience: 3+ years of experience in the job offered or a related occupation.</li><li>Technical Expertise: Must possess at least 3 years of hands-on experience in the following areas:<ul><li>Deep Learning &amp; Neural Networks</li><li>Machine Learning Pipeline Development</li><li>Sensor Fusion &amp; Signal Processing</li><li>Computer Vision &amp; Real-Time Inference Optimization</li><li>Statistical Modeling &amp; A/B Testing</li></ul></li></ul>Compensation &amp; Benefits<ul><li>Salary Range: $174,304 – $200,000 per year. (Final compensation is based on experience).</li><li>Full Medical, Dental, Vision, 401k</li><li>Unlimited PTO</li><li>Daily meals provided</li></ul><h2>How to apply</h2>Please send your resume and a cover letter to james@unitxlabs.com and ensure you include Ref#MM0426 in the subject line.

UnitX

Open jobs at UnitX

Machine Learning Engineer

Working for UnitX?

Featured companies