ML Engineer II, World Models

Diligent Robotics

4d

About The Position

We’re doing what isn’t easy, but nothing worth doing ever is.
We envision a future powered by robots that work seamlessly with human teams. We build artificial intelligence that enables service robots to collaborate with people and adapt to dynamic human environments. Join our mission-driven team as we build out current and future generations of robots.
As an ML Engineer II (World Models), you will develop and deploy predictive perception (“world model”) systems that fuse multi-sensor robot data into a unified representation of the near future. You’ll build training and evaluation workflows that convert fleet data into reliable model improvements, and partner with robotics engineers to ship the model on edge hardware. Your work will directly impact Moxi’s ability to move confidently and safely in crowded hospital environments.

Requirements

• Bachelor’s or Master’s degree in Robotics, Computer Science, Electrical Engineering, or related field (PhD a plus).
• 3+ years of experience building and training deep learning models in robotics, autonomy, or perception.
• Strong proficiency with PyTorch and modern training workflows (distributed training, mixed precision, profiling).
• Experience working with multimodal sensor data (cameras + LiDAR/depth) and temporal models.

Nice To Haves

• Experience with predictive perception / world models / video prediction.
• Experience deploying ML to edge devices (TensorRT/ONNX, quantization/INT8, runtime profiling).
• Familiarity with ROS pipelines, sensor calibration, and autonomy stack integration.
• Experience with simulation-based evaluation (Isaac Sim/Mujoco or similar) and offline replay testing.

Responsibilities

• Develop multimodal world-model architectures that ingest and fuse camera, LiDAR/depth, and robot state and produce short-horizon predictions.
• Build and maintain training pipelines: dataset construction, tokenization/backbones, distributed training, and ablation frameworks.
• Define model evaluation metrics and regression suites that reflect real robot outcomes.
• Create visualization/debug tooling for temporal predictions (rollouts, replays, overlays, failure case inspection).
• Optimize and distill models for edge deployment; benchmark latency, memory, and stability on target hardware.
• Collaborate with the AI Platform team to integrate the world model into autonomy stacks and validate behavior.
• Work with Operations to identify failure modes in the field and drive data curation and model iteration.

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level