United States
Roku
UnitX
Machine Learning Engineer
United States
Milpitas, California, United States
Go
Job Title
Machine Learning Engineer
How to apply
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
More opportunities with one profile
Want to be considered for more roles without repeating the whole process? Create your profile and be found by companies using Nort.
Create my profile