Research Engineer, World Models

About 1X
We’re an AI and robotics company based in Palo Alto, California, on a mission to build a truly abundant society through general‑purpose robots capable of performing any kind of work autonomously.
We believe that to truly understand the world and grow in intelligence, humanoid robots must live and learn alongside us. That’s why we’re focused on developing friendly home robots designed to integrate seamlessly into everyday life.
We’re looking for curious, driven, and passionate people who want to help shape the future of robotics and AI. If this mission excites you, we’d be thrilled to hear from you and explore how you might contribute to our journey.

Role Overview
You will build large multi‑modal generative “world models” that predict future sensor inputs and actions of the robot from its past observations. These foundational models are meant to enable the robot to understand and act in unstructured, real‑world environments. You will work across data pipelines, model architecture, and product deployment to drive improvements in robot autonomy.

Responsibilities

• Full‑stack engineering: data engineering, model architecture design, and delivering polished products

• Develop high‑throughput data loaders for large multi‑modal datasets

• Implement tokenizers and transformers tailored for web‑scale robot data

• Translate improvements in world model architectures into improvements in robot autonomy

• Predict how real‑world robot performance scales with pre‑training metrics (e.g., log loss)

• Strong experience with Python (and related tooling such as Bazel)

• Proficiency with PyTorch or equivalent deep learning frameworks

• Familiarity with simulation platforms (e.g. Isaac Sim, MuJoCo)

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

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

• Understanding of scaling laws and metrics for foundation models
  

Benefits & Compensation

• Salary Range: [Not publicly stated]

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