899,872open jobs
55,626companies
151,463added this week
Browse all
Salary
$160k – $180k per year
Location
In office (Milpitas)
Seniority
Junior · 2+ years exp
Employment
Full-Time

Confirmed on the employer's own hiring board on Sep 28, 2026. First seen by Alion on Apr 27, 2026. UnitX scores C on the Alion truth index.

Overview
Company
Impact
Profile match
Deploy high-accuracy in-line defect detection for Automotive, EV Battery, and Electronics production in days.

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:

  • 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.

Who You Are:

  • 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.

Preferred Qualifications:

  • Master’s degree in Computer Science, Mathematics, Physics, or a relevant technical field.

  • Experience with model inference optimization

  • Experience with non-ML CV algorithms

Compensation & Benefits

  • Competitive cash and equity

  • Full Medical, Dental, Vision, 401k

  • Unlimited PTO

  • Daily meals provided

Free account
Stop reading job ads. Get the ones that fit.
One free account turns this page into a shortlist built around your stack, your level and your pay.
Match on every job. Stack, seniority, pay and location, scored against your profile.
899,872 open roles. Read straight off company career pages, refreshed every day.
Unlimited applications. Every one you send is tracked in one place, on-site or on a company board.
3 tailored CVs a month. Rewritten for the exact job you are applying to. Included free.
Create a free account Continue with Google
Free forever. No card. Under a minute.

Your match

How well do you fit this role?
Two answers are enough for a real match. No account needed.
Check my fit
Answers stay in this browser until you create an account.

Recommended for you based on this role

AI/ML
Similar stack
Same company
Milpitas
$91k – $151k per year • In office • Full-Time • 3+ years exp • Bachelor's Degree • Waltham
Python
SQL
Databases
Microsoft Fabric
AI/ML
AI Agents
RAG
Copilot Studio
DevOps
Azure
Analytics
Power BI
Apply
≈ $113k – $215k per year (Estimated) • In office • Full-Time • 3+ years exp • Bachelor's Degree • Plano
Python
Java
Databases
Redis
Cassandra
ElasticSearch
OpenSearch
AI/ML
Embeddings
Prompt Engineering
RAG
Semantic Search
Semantic Search
DevOps
Azure
CI/CD
Git
AWS
Docker
Kubernetes
Management
Agile
Apply
$50k per year • In office • Internship • PhD • Miramar
Databases
Databricks
AI/ML
AI Agents
DevOps
Azure
AWS
Management
Agile
Microsoft Office
Apply
$138k per year • In office • 3+ years exp • Bachelor's Degree • Miramar
Python
SQL
Visual Basic
Databases
Databricks
AI/ML
AI Agents
OpenAI
Amazon SageMaker
DevOps
Azure
AWS
Analytics
Tableau
Power BI
SAP BusinessObjects
Management
Power Automate
SharePoint
Apply
ML Engineer 6 hours ago
up to $120k per year • In office • 1+ year exp • Atlanta
Python
AI/ML
Fine-tuning
TensorFlow
PyTorch
LLM
RAG
Machine Learning
DevOps
GCP
Azure
CI/CD
AWS
Docker
Kubernetes
IAM
Management
Agile
Apply
Data Scientist 2 days ago
≈ $57k – $130k per year (Estimated) • Remote (likely United States, US time zones hours) • 4+ years exp
Python
AI/ML
Claude
Prompt Engineering
Function Calling
AI Agents
Gemini
LLM
OpenAI
Anthropic
LLM Guardrails
Agentic Workflows
Machine Learning
DevOps
Azure
Apply
$75k – $95k per year • Remote (United States) • 20+ years exp • Bachelor's Degree
Python
SQL
Analytics
Power BI
Microsoft Excel
Apply
≈ $80k – $167k per year (Estimated) • In office • 5+ years exp • Bachelor's Degree • Vancouver
Python
C#
C++
AI/ML
Machine Learning
Game Dev
Unity
RenderDoc
Design
Adobe Photoshop
Figma
Cinema 4D
Apply
In office • 1+ year exp • Bachelor's Degree • Vancouver
Python
C#
Game Dev
Unity
Design
Adobe Photoshop
Apply
≈ $84k – $176k per year (Estimated) • In office • 5+ years exp • Bachelor's Degree • Vancouver
Python
C#
C++
AI/ML
Machine Learning
Game Dev
Unity
RenderDoc
Apply
$170k – $200k per year • In office • Full-Time • 3+ years exp • Bachelor's Degree • Milpitas
AI/ML
Stable Diffusion
Reinforcement Learning
Multimodal AI
Computer Vision
Segment Anything
Self-Supervised Learning
Explainable AI
Interpretability
Physical AI
Machine Learning
Robotics
Sensor Fusion
Reinforcement Learning
Analytics
A/B Testing
Apply
Embodied AI Engineer 5 months ago
$110k – $180k per year • In office • Full-Time • Master's Degree • Milpitas
Python
C++
C++
TensorFlow C++
PyTorch C++
AI/ML
CUDA Toolkit
Reinforcement Learning
JAX
Multimodal AI
TensorRT
TensorFlow
PyTorch
Synthetic Data
CUDA
Vision-Language-Action
Physical AI
Embodied AI
Machine Learning
Robotics
Drake
Isaac Sim
MuJoCo
Sim-to-Real
Imitation Learning
Reinforcement Learning
Apply
Solutions Manager 27 days ago
$170k – $200k per year • In office • Full-Time • 7+ years exp • Bachelor's Degree • Milpitas
Apply
$160k – $200k per year • In office • Full-Time • 7+ years exp • Milpitas
Apply
$140k – $150k per year • In office • Full-Time • 5+ years exp • Milpitas
AI/ML
Physical AI
Apply
≈ $65k – $130k per year (Estimated) • In office • 5+ years exp • Milpitas
Apply
Detail Technician 9 hours ago
≈ $43k – $103k per year (Estimated) • In office • Milpitas
Apply
Night Auditor 1 day ago
≈ $81k – $212k per year (Estimated) • In office • High School Diploma • Milpitas
Apply
≈ $81k – $212k per year (Estimated) • In office • Part-Time • PhD • Milpitas
Apply
$84k – $94k per year • In office • Contractor • Milpitas
Apply
See all jobs
This is one of many
899,872 more open roles from verified company boards, updated every day.