Senior/Staff ML Researcher - Generative AI Worlds in Brooklyn, NY

About the Role

We’re looking for an exceptional machine learning researcher to join our founding team. You’ll be pushing the frontier of diffusion transformer & world model research, solving core problems of how to create a universal simulator and how to integrate it into customers’ stacks to solve real-world problems.

Responsibilities

• Develop state-of-the-art generative models
• Read, implement and improve the latest papers in this field
• Rapid prototyping of ideas, approaches, methods etc.

Ideally you

• Have 3+ years of machine learning engineering/research experience
• Track record of implementing research papers in PyTorch
• Solid understanding of state-of-the-art generative models (VAEs, GANs, Autoregressive, Diffusion, etc.)
• Knowledge of computer graphics fundamentals (rendering, geometry processing, etc.)

We’re big believers that small, in-person teams are the ones that create the future. We work out of our Brooklyn, New York office, but are flexible with hybrid work schedules. Ideally you are based in NYC.