AI Research Engineer, Models

Overview

AI Research Engineer, World Models at 1X — Palo Alto, CA (on-site). We build humanoid robots that work alongside people to solve labor shortages and create abundance.

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.

• 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

• 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 Mu Jo Co
• 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

• Salary Range: $180,000 – $300,000 + Equity
• Health, dental, and vision insurance
• 401(k) with company match
• Paid time off and holidays

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.