Machine Learning Engineer
HQ · Milpitas, California, United States
- Pay
$174,304–200,000/year — pay source
Compensation & Benefits Salary Range: $174,304 – $200,000 per year. (Final compensation is based on experience). Full Medical, Dental, Vision, 401k
Read the full posting- Work setup
On-site — work setup source
Location type: On-site
From the employer’s posting- Employment
Full-time — employment source
Employment type: Full-time
From the employer’s posting- Team
Software — team source
Department: Software
From the employer’s posting
Before you apply
- Sponsorship
Visa sponsorship not confirmed — sponsorship source
Will you now or in the future require sponsorship for employment visa status (e.g., H-1B, TN, O-1)?
From the employer’s application form
What you’ll work on
Full postingBuild Automated Pipelines: Develop and maintain automated pipelines for data preprocessing, augmentation, and model training to continuously optimize deployed solutions.
You will optimize model inference for real-time decision-making, ensuring processing times remain under 20 milliseconds.
Develop Evaluation & Interpretability Tools: Create frameworks to assess model accuracy, precision, recall, and efficiency.
From the employer’s posting
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.
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.
What you’ll bring
All qualificationsCore experience
- Experience: 3+ years of experience in the job offered or a related occupation.
Qualification wording
Experience: 3+ years of experience in the job offered or a related occupation.
About UnitX
UnitX builds the world's leading physical AI systems to automate repetitive visual tasks in factories.
In the employer’s words · Read in context
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.
Your next step
- Have your CV and examples of relevant work ready.
- Check the listed location, eligibility and core experience before starting.
Complete your application on jobs.ashbyhq.com. The employer’s form will show what is required.
Already applied? Track this application
Source & posting history
Source notes
Source excerptsSelected passages from the saved posting. Check the full description for conditions and exceptions.
- Pay
Compensation & Benefits Salary Range: $174,304 – $200,000 per year. (Final compensation is based on experience). Full Medical, Dental, Vision, 401k
- Location & working pattern
Milpitas, California, United States
Working pattern and location restrictions need checking in the full posting.
- Work authorization
No clear work-authorization passage found. Eligibility is unconfirmed.
- Status in our records
- Active
- First seen by us
- Jul 2, 2026
- Recorded sightings
- 42
- Last seen by us
- Oct 11, 2026
- Employer says posted
- Jun 30, 2026
These dates show when we found the listing. Check the employer’s website to confirm it is still accepting applications.
Report an errorSee how this role fits your experience
Add your resume to compare the role’s scope, tools and requirements with your experience.