Senior/Staff AI Research Scientist, Foundation Models

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
• 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 to enable natural human-robot communication
• Optimize and deploy models as production-grade solutions on RoboForce robotic platforms

Requirements

• PhD in Machine Learning, Robotics, or related field, or Master's with 4+ years of experience
• Proficiency in Python and deep learning frameworks like PyTorch or JAX
• Expertise in large foundation models (VLM, VLA, etc.)
• Strong understanding of world model architectures and action-conditioned generative modeling
• Understanding of multimodal models and modern ML architectures like transformers and diffusion models
• Ability to work in-office 5 days per week in Milpitas, CA

Preferred Qualifications

• Experience with video generation or prediction models
• Strong publication record at top conferences (NeurIPS, ICML, CVPR, etc.)
• Expertise in neural network deployment (TensorRT) and GPU programming with CUDA
• Ability to design scalable experimentation and data pipelines

Benefits

• Competitive stock options and equity programs
• Health, dental, and vision insurance
• 401(k) plan
• Visa sponsorship and green card support
• Lunches, dinners, and a fully stocked kitchen

About the Company

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.