AI Research Engineer, World Models

AI Research Engineer, Foundation Models
Palo Alto, CA (on-site)

We build humanoid robots that work alongside people to solve labor shortages and create abundance.

The Role
You will build large multi-modal generative “world models” that predict future sensor inputs and robot actions based on past observations. These foundational models enable robots to understand and operate effectively in unstructured, real-world environments. You’ll work across data pipelines, model architectures, and deployment to drive improvements in robot autonomy. This role combines cutting-edge research with hands-on product development to push the limits of robot intelligence.

You Will

• Build full-stack systems across data engineering, model architecture, and product delivery

• Develop high-throughput data loaders for large-scale, multi-modal robot datasets

• Implement tokenizers and transformers designed for web-scale robotic data

• Improve robot autonomy by advancing world model architectures

• Predict robot performance in the real world based on pre-training metrics such as log loss

Must Have

• Strong experience with Python and related tools such as Bazel

• Proficiency with PyTorch or equivalent deep learning frameworks

• Familiarity with simulation platforms such as Isaac Sim or MuJoCo

• Experience working with multi-modal generative models combining video, audio, text, and action prediction

• Ability to design and optimize large-scale data pipelines for training

• Understanding of scaling laws and performance metrics for foundation models

Benefits & Compensation

• Salary Range: $180,000 – $300,000

• Health, dental, and vision insurance

• 401(k) with company match

• Paid time off and holidays

Equal Opportunity Employer
1X is an Equal Opportunity Employer. All qualified applicants will receive consideration for employment without regard to race, color, religion, sex, gender, gender identity or expression, sexual orientation, national origin, ancestry, citizenship, age, marital status, medical condition, genetic information, disability, military or veteran status, or any other characteristic protected under applicable federal, state, or local law.