• Companies
  • News
By technology
  • Python17.1k
  • AWS12k
  • Kubernetes8.7k
  • Docker8.3k
  • C++5.7k
  • PostgreSQL5.4k
  • SQL5.2k
By category
  • Backend Developer11.5k
  • DevOps Engineer3.4k
  • Machine Learning Engineer3.3k
  • Full Stack Developer3.2k
  • Data Engineer2.8k
  • Embedded Engineer2.2k
  • Frontend Developer862
By city
  • San Francisco, CA6.2k
  • New York, NY5.6k
  • Seattle, WA1.3k
  • Washington, DC1.2k
  • Toronto, ON1.2k
  • Boston, MA1k
  • Austin, TX999
Work mode
  • On-site22k
  • Remote10.2k
  • Hybrid7.3k
Job type
  • Full-time38.2k
  • Contract600
  • Part-time529
  • Freelance215
  • internship4
  • seasonal1
Experience
  • Entry level
  • Mid level
  • Senior
  • Staff+
  • Engineering management

The job board for developers. Find your next role at companies that value great engineering.

© 2023-2026 . All rights reserved.

Logos provided by Logo.dev
All jobs

AI Research Scientist

Embedding VC · Milpitas, CA

Extracted from the posting

Base
Not stated
Remote
5 days/week in-office
Visa
Visa sponsorship and green card support
Stack
Python, PyTorch, JAX, CUDA

AI Research Scientist (Senior/Staff) on Foundation Models at RoboForce, building VLM/VLA models and action-conditioned world models for robot planning and manipulation. Work covers multimodal inputs, spatial reasoning, and deploying models on production robot platforms. Stack is Python and PyTorch or JAX, with TensorRT and CUDA as bonus skills.

Why RoboForce

RoboForce is an AI robotics company developing Physical AI–powered Robo-Labor for dull, dirty, and dangerous work. The company's robots are engineered for demanding industrial environments, with a focus on real-world deployment and scalability.

We are looking for a Senior / Staff AI Research Scientist, Foundation Models to advance robotic embodied intelligence. In this role, you will develop algorithms that enable robots to understand their environment, interpret and execute tasks, and communicate seamlessly with humans — with a particular focus on building and training world models that allow robots to predict, plan, and generalize across complex physical tasks.

Responsibilities

• Design and deploy vision-language(-action) models (VLM/VLA) for contextual understanding and generalized robot action policies.

• Develop and train world models for action-conditioned prediction, long-horizon planning, and environment simulation — enabling robots to reason about the consequences of their actions before execution.

• Research approaches to improve world model fidelity using multi-modal inputs including vision, language, proprioception, and spatial representations.

• Develop foundation models with spatial reasoning capabilities to achieve high-precision robotic actions.

• Integrate multi-modal data sources (vision, language, speech, etc.) to enable natural human-robot communication.

• Optimize and deploy models as production-grade solutions on RoboForce robotic platforms.

Requirements

• PhD degree in Machine Learning, Robotics, or related field, or Master's degree with 4+ years of relevant experience.

• Proficiency in Python and deep learning frameworks (e.g., PyTorch, JAX).

• Expertise in large foundation models (VLM, VLA, etc.).

• Strong understanding of world model architectures and action-conditioned generative modeling for robot learning.

• Decent understanding of multimodal models, modern ML architectures (transformers, diffusion models, etc.).

• Requires 5 days/week in-office collaboration with the teams.

Bonus Qualifications

• Experience with video generation or prediction models (e.g., diffusion-based video models, autoregressive video transformers) and their application to world modeling or synthetic data generation for robot learning.

• Strong publication record at top conferences (NeurIPS, ICML, CVPR, ICCV, CoRL, ICRA, or equivalent).

• Expertise in neural network deployment (e.g., TensorRT) and GPU programming with CUDA.

• Proven ability to design scalable experimentation and data pipelines.

Benefits

• Competitive stock options/equity programs.

• Health, dental, and vision insurance, 401(k) plan.

• Visa sponsorship and green card support for qualified candidates.

• Lunches and dinners, a fully stocked kitchen, and regular team-building events.

Technologies

PythonPyTorchJAXCUDATensorRT
Embedding VC
Embedding VC
Website

Job details

TypeFull-time
Work ModeOn-site
LevelSenior
Location🇺🇸Milpitas, CA
PostedNov 05, 2026
Apply Now

Related jobs

Crusoe

Staff Software Engineer, AI Model Lifecycle

Crusoe
$250,000 - $300,000🇺🇸 San Francisco, CA +1On-site · Full-timePython · Golang · PyTorch
SpaceXAI

Model Engineer, Multimodal Understanding

SpaceXAI
$160,000 - $250,000🇺🇸 Palo Alto, CAOn-site · Full-timePython · PyTorch · JAX
Samsara

Staff Machine Learning Engineer - Edge AI

Samsara
$178,640 - $319,000Remote - USRemote · Full-timePython · C++ · Rust