Staff ML Architect: World Models & Scale

Odyssey

Palo Alto, Northern (CA, KY)

Hybrid

USD 180,000 - 320,000

Full time

14 days+

Application generator

Turn this role into an interview — a resume and cover letter built around what this employer wants.

Get past ATS filters

Job summary

Odyssey in Palo Alto seeks an ambitious ML engineer to help build large foundation world models, spanning data pipelines, training at scale, and real-time inference. You will own a unified system, data, architecture, training, inference, and evaluation, pushing the limits of current capabilities.

The role requires hands-on experience training large models, multi-node scaling, and proficiency with PyTorch (or TF/JAX).

Qualifications

2+ years of software engineering experience with ML performance.
4+ years of ML engineering experience, or a PhD in a related field.
Hands-on experience training large models at multi-node scale.
Proficiency with PyTorch (or TF/JAX).
Holistic ML engineer, owning end-to-end pipelines from data to deployment.

Responsibilities

Learn what makes large real-time world models tick.
Run scaling studies and apply scaling laws to guide experiments.
Own world model training at scale across thousands of GPUs.
Implement state-of-the-art ML algorithms, define metrics, iterate on leaderboards.
Work end to end across data pipelines, training, inference, and evaluation.
Be part of a team shaping the world model space.
Exploit latest GPU features to boost training and inference efficiency.
Take ownership of the full ML stack and core frameworks.