Generative AI - 3D Foundation Model

About the role

You will drive the development of our core 3D-native generative foundational model. In this role, you will join our foundational research to advance 3D AI, apply learnings from other fields of ML, and push the state of the art.

Responsibilities

• Design, train, and refine large-scale 3D generative models from covering pre-training, post-training, and emerging paradigms in diffusion, flow matching, and multi-modal learning.
• Bridge the gap between cutting-edge research and product, deploy models in real products used by millions of creators, using human feedback and creative evaluation.
• Create novel model architectures to make 3D generation faster, higher-quality, and more controllable.
• Collaborate with infrastructure and systems teams to build scalable training, and data pipelines across GPU clusters and cloud environments.
• Bring engineering discipline into a fast-paced research environment: elegant code, reproducible experiments, and building software as a team.
• Share insights and breakthroughs through internal demos, open-source contributions, or technical reports that advance the field of 3D generative AI.

Requirements

Strong engineering skills in Python and deep learning frameworks (preferably PyTorch); comfortable moving between research prototypes and production systems.

Familiar with Transformers and modern generative AI models (Diffusion / flow matching, VAE, etc.).

Curiosity and passion for multi-modal AI, and have an intuitive understanding of how models perceive, represent, and generate 3D worlds.

Familiar with high performance training on large scale infrastructure (e.g., SLURM, Ray, k8s).

Contributions to popular open-source machine learning projects or publications in top-tier CV / ML conferences is a plus.

Qualifications

We are looking for someone who:

• Has a deep understanding of the transformer architecture.
• Is proficient in Python and tensor programming.
• Believes in "the scaling law" and is excited to contribute to advancing AI that understands and generates 3D worlds.
• Can translate high-level goals into concrete research and implementation steps, set an approach, and follow through.
• Brings clarity to complex technical issues and is not afraid of confronting new ideas.
• Is eager to share knowledge with the team and create real-world benefits for our researchers, engineers, and millions of users.

Skills

Translate high-level goals into concrete research and implementation steps.

Set an approach and follow through.

Bring clarity to complex technical issues.

Confront new ideas and share knowledge with the team.

Benefits

Enjoy a vibrant work environment, daily interactions with leadership, and all the perks of working in the heart of Silicon Valley.

Pay

Competitive salary based on experience.

Schedule

Full-time, remote option available.