Member of Technical Staff, Research

You will set and pursue research directions in diffusion and flow models, world models, action representations, latent dynamics, and embodied policy learning. You will advance Distributed Diffusion Models for physical AI, design rigorous benchmark experiments, analyze failures, communicate findings clearly, and collaborate with systems engineers to keep experiments reproducible and scalable across heterogeneous compute.

Responsibilities

• Set and pursue research directions across diffusion, world models, action representations, latent dynamics, and embodied policy learning.
• Advance Distributed Diffusion Models for physical AI through distributed world models, expert ensembles, routing, specialization, and compositional generalization.
• Explore latent action representations for embodied AI.
• Design rigorous experiments on physical AI benchmarks with baselines, metrics, and failure analysis.
• Communicate results clearly to researchers and infrastructure engineers.
• Collaborate with systems engineers to keep experiments reproducible and scalable across heterogeneous compute.

Requirements

• Strong research judgment in diffusion, flow, score models, world models, representation learning, robotics, simulation, imitation learning, reinforcement learning, or video and multimodal generation.
• Experiment design skills, including baselines, metrics, and failure modes.
• Fluency in reading and implementing recent machine learning papers.
• Clear technical writing.
• Strong Python skills and experience with a modern machine learning framework.

Benefits

• Meaningful equity.
• Paid travel to top machine learning conferences worldwide.