UnitX
Posted 178 days agoVerified live 1d ago
Machine Learning Engineer
Brief overview
Milpitas, CAIn-person
UndergradOr in progress
$160k–$180k/yrStated range
2+ yrsMinimum
14 H-1B approvalsDept. of Labor
Deep LearningPyTorchTensorflowPythonComputer VisionModel inference optimizationNon-ML computer vision algorithmsStable DiffusionSAM (Segment Anything Model)
About the company
UnitXunitxlabs.com
UnitX builds robots for factories.
Visa sponsorship history
4 years sponsoring, last filed FY2026
Data powered by U.S. Department of Labor. This does not guarantee sponsorship for this specific role.14H-1B approved
100%approval rate
2new H-1B hires
$172,500median wage / yr
H-1B Petition ApprovalsVisas USCIS actually granted: the strongest sign the company sponsors.
20237
20243
20253
20261
LCA Certified ApplicationsAn early filing step, not a visa approval: it signals intent, not confirmed sponsorship.
20234
Top sponsored roles
Senior Firmware EngineerSenior Software Engineer in TestSenior Product Design EngineerField Sales Engineer
Job description
Summary
UnitX builds the world's leading physical AI systems to automate repetitive visual tasks in factories. They are seeking a Machine Learning Engineer to develop algorithms for defect detection and improve manufacturing efficiency using computer vision technology.
Responsibilities
- You will develop new algorithms that consume raw sensor data input to detect defect segmentation and location information on high-resolution images or 3D inputs, with pixel-level precision
- You will engineer software that runs on the production line 24/7 to efficiently execute our algorithms in real time with each decision made under 20ms
- You will develop metrics and tools to evaluate various model performance and improve visibility and interpretation of our system
- You will explore new solutions and push the boundaries of the field including using Stable Diffusion, SAM and build applications for critical missions in the manufacture fields
Skills
- Bachelor's degree in Computer Science, Mathematics, Physics, or a relevant technical field, or equivalent practical experience
- Solid foundational and applied math knowledge
- 2+ years of experience in building machine learning models for computer vision in production
- Strong theoretical background and practical experience in Deep Learning, with proficiency in PyTorch or Tensorflow
- Excellent Python skills for writing efficient and maintainable solutions in large code bases
- Strong communication and decision-making skills
- Ability to explain the rationale behind experiments and make decisions to balance exploration and exploitation
- Resilience in uncertain and complex environments
- Master's degree in Computer Science, Mathematics, Physics, or a relevant technical field
- Experience with model inference optimization
- Experience with non-ML CV algorithms
Qualifications
Must Haves
- Bachelor's degree in Computer Science, Mathematics, Physics, or a relevant technical field, or equivalent practical experience
- Solid foundational and applied math knowledge
- 2+ years of experience in building machine learning models for computer vision in production
- Strong theoretical background and practical experience in Deep Learning, with proficiency in PyTorch or Tensorflow
- Excellent Python skills for writing efficient and maintainable solutions in large code bases
- Strong communication and decision-making skills
- Ability to explain the rationale behind experiments and make decisions to balance exploration and exploitation
- Resilience in uncertain and complex environments
Nice to Haves
- Master's degree in Computer Science, Mathematics, Physics, or a relevant technical field
- Experience with model inference optimization
- Experience with non-ML CV algorithms
Benefits
- Competitive cash and equity
- Full Medical, Dental, Vision, 401k
- Unlimited PTO
- Daily meals provided
