Robotics ML Engineer: Edge AI for Production Robots

Origin; San Francisco (CA); On-site; USD 120,000 - 160,000; Full time

Job summary

A cutting-edge robotics company is searching for a candidate to manage the full lifecycle of learned components on OG-1, focusing on training and deploying VLA models within a dynamic environment. Ideal candidates will possess a strong educational background in CS, Robotics, or ML and demonstrate proficiency in Python and PyTorch, as well as experience in real-world ML deployment. Responsibilities include optimizing models for edge performance and researching novel approaches for physical interactions in robotics. Competitive compensation offered.

Qualifications

• Strong Python and PyTorch; comfort modifying research codebases.
• Experience deploying ML on real hardware with debugging.
• Working knowledge of robotics middleware.

Responsibilities

• Train and deploy VLA models for contact-rich manipulation.
• Build teleoperation pipelines and optimize models for edge.
• Research and prototype world models for prediction.

Skills

Python

PyTorch

Imitation learning

Reinforcement learning

Vision-language models

Robot learning from demonstration

Sim-to-real

ROS2

GPU profiling and optimization

Education

BS/MS/PhD in Computer Science, Robotics, or Machine Learning

Tools

Jetson AGX Orin

TensorRT

ROS2

CUDA

Isaac Sim

Job description

A cutting-edge robotics company is searching for a candidate to manage the full lifecycle of learned components on OG-1, focusing on training and deploying VLA models within a dynamic environment. Ideal candidates will possess a strong educational background in CS, Robotics, or ML and demonstrate proficiency in Python and PyTorch, as well as experience in real-world ML deployment. Responsibilities include optimizing models for edge performance and researching novel approaches for physical interactions in robotics. Competitive compensation offered.