Jobs in Milpitas, CA

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

What you'll bring

• MS or Ph.D. in Computer Science, Robotics, Machine Learning, Artificial Intelligence, or a closely related quantitative field.
• Strong, production-level coding skills with deep expertise in Python and C++.
• 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.
• 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.
• A demonstrated track record of taking complex, open-ended physical challenges and translating them into elegant, mathematically grounded algorithmic solutions.
• Familiarity with deploying, profiling, and optimizing ML models for real-time inference on robotic hardware (e.g., NVIDIA Jetson, TensorRT, CUDA).
• Experience exploring or applying large-scale generative models, Vision-Language-Action (VLA) models, or Foundation Models to robotic planning and control.
• Proficiency with industry-standard 3D asset and robotics description formats (e.g., USD, URDF, MJCF).

Benefits