Research Scientist - World Models

SpAItial AI; Greater London; On-site; GBP 80,000 - 100,000; Full time

Job summary

SpAItial AI is seeking innovative individuals to pioneer the next generation of World Models in London. You'll architect generative world models and develop foundational ML models that push the boundaries of generative AI and computer vision.

Ideal candidates hold a PhD in Computer Science or a related field, possess strong knowledge of generative models and deep learning frameworks, and excel in coding and ML experimentation. Join us to shape the future of AI!

Qualifications

• PhD with top-tier publication record in machine learning, vision, or graphics.
• Strong knowledge of diffusion models and architectures.
• Experience with coding, rapid ML experimentation, and AI tools.

Responsibilities

• Architect generative world models considering space, time, and physics.
• Design and develop foundational ML models for image/video diffusion.
• Conduct large-scale model training on distributed cloud infrastructure.
• Develop demos to showcase trained model prototypes.
• Process and maintain large datasets for training.
• Optimize models during production and fine-tuning at test time.

Skills

Generative models expertise

Deep learning frameworks (e.g., PyTorch)

Expert coding skills

Education

PhD in Computer Science or related field

Job description

SpAItial is pioneering the next generation of World Models, pushing the boundaries of generative AI, computer vision, and the simulation of reality. We are moving beyond 2D pixels to build models that natively understand the physics and geometry of our world. Our mission is to redefine how industries, from robotics and AR/VR to gaming and cinema, generate and interact with physically-grounded 3D environments.

We’re looking for individuals who are bold, innovative, and driven by a passion for pushing the boundaries of what’s possible. You should thrive in an environment where creativity meets challenge and be fearless in tackling complex problems. Our team is built on a foundation of dedication and a shared commitment to excellence, so we value people who take immense pride in their work and place the collective goals of the team above personal ambition. As a part of SpAItial, you’ll be at the forefront of the AI revolution in generative AI technology, and we want you to be excited about shaping the future of this dynamic field. If you’re ready to make an impact, embrace the unknown, and collaborate with a talented group of visionaries, we want to hear from you.

Responsibilities

• Architect generative world models that reason about space, time, and physics.
• Design and develop image/video diffusion foundational ML models.
• Large-scale distributed model training on top of distributed cloud infra.
• Develop demos showcasing the trained model prototypes.
• Processing and maintaining large data for model training.
• Productionizing models, test-time model optimization.

Key Qualifications

• PhD in Computer Science or related field with a top-tier publication record in ML, Vision or graphics (CVPR, ECCV/ICCV, NeurIPS, SIGGRAPH, etc.).
• Strong knowledge of generative models such as image or video diffusion.
• Strong knowledge of cutting-edge architectures including diffusion transformers (DiT).
• Rich experience with deep learning frameworks such as PyTorch.
• Expert coding skills, and ability to rapidly iterate through ML experiments, including usage of modern AI coding tools.

At SpAItial, we are committed to creating a diverse and inclusive workplace. We welcome applications from people of all backgrounds, experiences, and perspectives. We are an equal opportunity employer and ensure all candidates are treated fairly throughout the recruitment process.