Diligent Robotics

Robots + humans working together as a team.

ML Engineer II, World Models

Machine Learning EngineerMachine Learning EngineerFull TimeRemoteTeam 1-10Since 2017H1B SponsorCompany SiteLinkedIn

Location

United States

Posted

10 days ago

Salary

Not specified

Bachelor Degree3 yrs expEnglishPy Torch

Job Description

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

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

Related Job Pages

More Machine Learning Engineer Jobs

Machine Learning Engineer10 days ago
Full TimeRemoteTeam 11-50Since 2022

The role involves leading the design, implementation, and scaling of the company's ML infrastructure and production AI systems across the entire model lifecycle. This includes architecting, building, and scaling the end-to-end ML Ops pipeline, from training and evaluation to deployment and monitoring.

PythonTypeScriptGoPyTorchTransformersvLLMCUDADockerKubernetesHelmAWSGCPAzureML Opsdistributed systemsGPU accelerationmodel deploymentCI/CDmodel monitoringfeature pipelines
United States
Machine Learning Engineer10 days ago
Full TimeRemoteTeam 1,001-5,000

The role involves building and maintaining infrastructure on AWS, Terraform, and Kubernetes to support AI/ML, including Generative AI applications at scale. Responsibilities also include managing the end-to-end lifecycle of ML models, ensuring observability, and developing tooling to streamline model development and deployment for AI/ML Scientists.

AWSKubernetesJavaPythonC++Terraform
United States
Machine Learning Engineer10 days ago
Full TimeRemoteTeam 1,001-5,000Since 2010H1B No Sponsor

Senior ML Ops Engineer maintaining AI/ML infrastructure at Sprout Social

AWSJavaKubernetesPythonTerraform
United States
$135.7K - $205.3K / year

Staff Machine Learning Engineer

Angi

Your home for everything home.

Machine Learning Engineer10 days ago
Full TimeRemoteTeam 1,001-5,000Since 2016H1B Sponsor

Staff Machine Learning Engineer at Angi developing impactful algorithms for matchmaking and search ranking.

CloudPythonSQL
New York
$200K - $280K / year