Demo

Senior Embodied AI Engineer

UnitX
Milpitas, CA Full Time
POSTED ON 4/14/2026
AVAILABLE BEFORE 5/9/2026
Title:

Senior Embodied AI 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.

The Role

We are seeking a visionary and driven Embodied AI Engineer to serve as a cornerstone of our intelligent robotics team. In this role, you will own the critical intersection of virtual environments and physical intelligence. You will not only architect the high-fidelity physics simulations that serve as the virtual proving grounds for our fleet, but also design, train, and deploy the advanced machine learning paradigms that learn from them. By controlling this entire sim-to-real lifecycle—seamlessly turning massive amounts of synthetic and real-world data into robust, scalable algorithms—you will empower our robots to generalize their intelligence across a multitude of complex, high-precision manufacturing tasks.

Key Responsibilities

  • Architect Intelligent Control Systems: Design, build, and deploy sophisticated, data-driven control architectures. You will create frameworks that synthesize complex, multi-modal sensor inputs, enabling our robotic systems to make nuanced, real-time decisions in dynamic environments.
  • Bridge the Sim-to-Real Divide: Champion our core data and simulation strategies. Construct highly accurate, physics-based simulation environments to identify and mitigate real-world failure modes.
  • Build Robust Data Flywheels: Harmonize massive datasets by seamlessly blending unstructured, messy real-world sensor data with task-oriented synthetic data to fuel the continuous training and refinement of our foundation models.
  • Pioneer Adaptive Behaviors: Drive the application of advanced machine learning methodologies (such as Reinforcement Learning, Imitation Learning, and Behavior Cloning) to physical systems, developing generalized robotic skills that seamlessly transition across diverse manufacturing applications.
  • Scale for Global Impact & Edge Deployment: Ensure that all algorithmic solutions are rigorously evaluated, diagnosed, and optimized for rapid deployment and massive scalability. You will deploy these models directly to edge compute devices on our global factory network, maintaining uncompromised precision, low latency, and reliability.

Basic Qualifications

  • Education & Experience: MS or Ph.D. in Computer Science, Robotics, Machine Learning, Artificial Intelligence, or a closely related quantitative field.
  • Programming Excellence: Strong, production-level coding skills with deep expertise in Python and C .
  • Machine Learning Mastery: Deep expertise in developing modern machine learning architectures using leading deep learning frameworks (e.g., PyTorch, JAX, TensorFlow), managing the full lifecycle from data ingestion to deployment.
  • Simulation Expertise: Proven ability to leverage high-fidelity physics engines (e.g., NVIDIA Isaac Sim, MuJoCo, Drake) to train, evaluate, and validate AI models prior to real-world deployment.
  • Visionary Problem Solving: A demonstrated track record of taking complex, open-ended physical challenges and translating them into elegant, mathematically grounded algorithmic solutions.

Preferred Qualifications

  • Edge Compute Experience: Familiarity with deploying, profiling, and optimizing ML models for real-time inference on robotic hardware (e.g., NVIDIA Jetson, TensorRT, CUDA).
  • Advanced Embodied AI: Experience exploring or applying large-scale generative models, Vision-Language-Action (VLA) models, or Foundation Models to robotic planning and control.
  • 3D/Robotics Toolchains: Proficiency with industry-standard 3D asset and robotics description formats (e.g., USD, URDF, MJCF).

Benefits:

  • Competitive salary & equity
  • Unlimited PTO
  • Full Medical, Dental, Vision, 401k
  • Daily meals provided with your own choice

Salary : $15

If your compensation planning software is too rigid to deploy winning incentive strategies, it’s time to find an adaptable solution. Compensation Planning
Enhance your organization's compensation strategy with salary data sets that HR and team managers can use to pay your staff right. Surveys & Data Sets

What is the career path for a Senior Embodied AI Engineer?

Sign up to receive alerts about other jobs on the Senior Embodied AI Engineer career path by checking the boxes next to the positions that interest you.
Income Estimation: 
$119,030 - $151,900
Income Estimation: 
$149,493 - $192,976
Income Estimation: 
$101,387 - $124,118
Income Estimation: 
$119,030 - $151,900
Employees: Get a Salary Increase
View Core, Job Family, and Industry Job Skills and Competency Data for more than 15,000 Job Titles Skills Library

Job openings at UnitX

  • UnitX Milpitas, CA
  • Job Title: Solutions Architect About Us: UnitX is building the world’s best robotics product to accelerate human productivity in manufacturing. UnitX is a ... more
  • 13 Days Ago

  • UnitX Milpitas, CA
  • Job Title: Senior HR Manager About Us: UnitX builds the world's leading physical AI systems to automate repetitive visual tasks in factories. UnitX is a fa... more
  • 14 Days Ago

  • UnitX Milpitas, CA
  • Title: Senior Simulation Engineer About Us: UnitX builds the world's leading physical AI systems to automate repetitive visual tasks in factories. UnitX is... more
  • 5 Days Ago

  • UnitX Milpitas, CA
  • Job Title: Operations Associate About Us: UnitX builds the world's leading physical AI systems to automate repetitive visual tasks in factories. UnitX is a... more
  • 8 Days Ago


Not the job you're looking for? Here are some other Senior Embodied AI Engineer jobs in the Milpitas, CA area that may be a better fit.

  • Intuitive Sunnyvale, CA
  • Company Description It started with a simple idea: what if surgery could be less invasive and recovery less painful? Nearly 30 years later, that question s... more
  • 1 Day Ago

  • Feather Robotics Palo Alto, CA
  • About the Job We're seeking an engineer passionate about advancing general-purpose labor through robotics. You'll play a central role in developing the AI ... more
  • 10 Days Ago

AI Assistant is available now!

Feel free to start your new journey!