Senior AI Research Scientist – Robotic World Models

Embedding VC; Milpitas (CA); On-site; USD 180,000 - 280,000; Full time

Benefits offered by this job

Stock options

Health insurance

Visa sponsorship

Lunches and team events

Job summary

RoboForce in Milpitas, CA is seeking a Senior/Staff AI Research Scientist to advance embodied intelligence in robotic systems. You will design and train foundation models that enable robots to understand environments, plan actions, and interact with humans.

Responsibilities include building vision-language-action models, world models for long-horizon planning, and integrating multi-modal data. PhD or MS+4 years, Python and PyTorch/JAX, and in-office collaboration 5 days/week are required.

Qualifications

• PhD or MS with relevant robotics/ML research.
• Strong Python and DL framework skills (PyTorch, JAX).
• Experience with foundation models (VLM/VLA) and world models.
• Good understanding of multimodal models and modern ML architectures.
• Requires 5 days/week in-office collaboration.

Responsibilities

• Design and deploy vision-language models (VLM/VLA) for contextual understanding and robot actions.
• Develop and train world models for action-conditioned prediction, planning, and environment simulation.
• Research improvements to world model fidelity using vision, language, proprioception, and spatial representations.
• Develop foundation models with spatial reasoning for high-precision robotic actions.
• Integrate multi-modal data to enable natural human-robot communication.
• Optimize and deploy models as production-grade solutions on RoboForce platforms.

Skills

Vision-language models

World models

Multimodal learning

Robotics expertise

Transformers/Diffusion

Education

PhD in ML/Robotics

MS in related field +4 years

Tools

PyTorch

JAX

TensorRT

CUDA

Job description

RoboForce in Milpitas, CA is seeking a Senior/Staff AI Research Scientist to advance embodied intelligence in robotic systems. You will design and train foundation models that enable robots to understand environments, plan actions, and interact with humans.

Responsibilities include building vision-language-action models, world models for long-horizon planning, and integrating multi-modal data. PhD or MS+4 years, Python and PyTorch/JAX, and in-office collaboration 5 days/week are required.