Member of Technical Staff, Diffusion World Models & Robotics
About this role
Who we are
Odyssey is an AI lab pioneering general world models: causal, multimodal systems that learn to predict and interact with the world over long horizons. This foundational technology promises to revolutionize robotics, science, healthcare, education, gaming, defense, and beyond.
Odyssey’s founders previously pioneered the most complex application of physical AI: self-driving cars. They’ve now brought together a world-class research team from DeepMind, Tesla, Waymo, Meta, Apple, and Wayve, who have made significant contributions to language models (DeepMind Gemini), video models (DeepMind Veo), world models (Wayve GAIA), and autonomous systems (Tesla FSD).
Odyssey has raised significant venture capital from GV, Amazon, AMD, EQT, NVIDIA, Natural Capital, In-Q-Tel, Elad Gil, Jeff Dean, Guillermo Rauch, Garry Tan, Kyle Vogt, and researchers from OpenAI, DeepMind, MSL, Recursive, and Thinking Machines.
What we're looking for
We are looking for people with a deep interest in improving machine learning algorithms at the intersection of diffusion world models and robotics. Interactive, action-conditioned world models are a cutting-edge research area that is not yet mature. You will be working at the cusp of what’s possible, using world models as simulators, feature extractors, and training grounds for real robot policies and model-based RL approaches. Most new experiments here will fail; your focus will be on maximally learning from failed experiments to increase the chances of eventual success on real hardware.
What you'll do
Learn what makes interactive world models tick. How data, diffusion backbones, action conditioning, and downstream policies interact.
Implement state-of-the-art diffusion and robot-learning algorithms, and design losses, reward functions, and reinforcement / preference-based fine-tuning for interactivity and policy performance.
Run a high cadence of ablations across WM and policy, from architecture and feature taps to data mixes and conditioning schemes.
Build the bridge to real robots: data pipelines, WM fine-tuning, and policy deployment onto Robots-aaS and design partner systems (remote and face-to-face).
Exploit the latest features on modern GPUs to increase training and inference efficiency.
Take ownership of the full ML stack, including core frameworks that Odyssey researchers and product engineers rely on.
Who you are
A PhD (or equivalent research experience) plus 2+ years of relevant research or engineering experience, or 4+ years of software engineering experience with 2+ years of relevant ML work.
Significant hands-on experience with one or more of: diffusion models, world models, model-based RL (e.g. Dreamer lineage), or vision-language-action (VLA) policies.
Track record of owning projects end to end.
Not shy to touch any stage of an ML pipeline, from data to real-robot deployment; sim-to-real experience is a strong plus.
Proficiency with PyTorch (or TF/JAX).
Highly experiment driven.
Flexible to work in-person in Zurich, London, or the Bay Area. Most of the team works in-person in Zurich at the moment.
Company at a glance
Odyssey is a technology-driven company dedicated to helping families access state funding to attend the school of their choice. It partners with state agencies to administer school choice programs, including education savings accounts (ESAs), and its technology powers programs that serve more than 115,000 students across the United States with over $225 million in state funding. Sitting at the intersection of technology, government, and education, Odyssey collaborates with K-12 providers and vendors to create a robust marketplace for families. Backed by well-known investors and advisors, Odyssey is gaining traction quickly with ample opportunity for growth. The company is expanding its team and is seeking top-tier talent to join its mission; for careers, contact [email protected].
Tired of cold applications?
Sign up with Clera and we'll reach out the moment a role actually fits you — no more spraying applications into the void.
Know someone who'd be great for this?