Staff ML Engineer - Real-Time World Models

Odyssey; Palo Alto (CA); On-site; USD 180,000 - 240,000; Full time

Job summary

Odyssey is an AI lab pioneering general world models and advancing real-time ML research. You will work at the cusp of what is possible, tackling experiments that fail to learn and focusing on deriving maximum knowledge from them to drive future success.

You will implement state-of-the-art ML algorithms, define robust metrics, and relentlessly iterate on leaderboards while exploiting GPU features to boost training and inference efficiency.

Qualifications

• 1+ years of software engineering experience with ML focus.
• 2+ years of ML experience.
• Track record of owning projects end to end.
• Not shy to touch any stage of an ML pipeline.
• Proficiency with PyTorch (or TF/JAX).
• Highly experiment driven.

Responsibilities

• Learn what makes real time world models tick and how data, models, and diffusion interact.
• Implement state-of-the-art ML algorithms and define metrics on leaderboards.
• Be part of a team defining and leading the world model space.
• Exploit latest GPU features to increase training and inference efficiency.
• Take ownership of the full ML stack, including core frameworks.

Skills

ML performance

End-to-end ownership

Experimentation

Software engineering

Tools

PyTorch

TensorFlow/TF-JAX

Job description

Odyssey is an AI lab pioneering general world models and advancing real-time ML research. You will work at the cusp of what is possible, tackling experiments that fail to learn and focusing on deriving maximum knowledge from them to drive future success.

You will implement state-of-the-art ML algorithms, define robust metrics, and relentlessly iterate on leaderboards while exploiting GPU features to boost training and inference efficiency.