Research Scientist (Generative Modeling) San Francisco

World Labs Inc.; San Francisco (CA); On-site; USD 250,000 - 325,000; Full time

Benefits offered by this job

Equity awards

Annual performance bonus

Job summary

World Labs Inc. in San Francisco is seeking a Research Scientist specializing in generative modeling, particularly diffusion models. This role focuses on developing large-scale models and collaborating with cross-functional teams to translate product requirements into effective applications.

The ideal candidate has over 3 years of experience in machine learning and strong expertise with frameworks like PyTorch or TensorFlow. The position offers a salary ranging from $250,000 to $325,000 plus benefits and equity.

Qualifications

• 3+ years of experience in generative modeling or applied ML roles.
• Extensive experience with machine learning frameworks such as PyTorch or TensorFlow.
• Strong coding proficiency in Python and experience with GPU-accelerated computing.
• Comfortable in a dynamic startup environment with high levels of ambiguity.

Responsibilities

• Design, implement, and train large-scale diffusion models for generating 3D worlds.
• Collaborate closely with research and product teams to translate requirements into technical roadmaps.
• Mentor colleagues and drive best practices in generative modeling and ML engineering.

Skills

Generative modeling

Machine learning frameworks (PyTorch, TensorFlow)

Python coding proficiency

3D vision

Tools

GPU-accelerated computing

Large-scale training infrastructure

Job description

Research Scientist (Generative Modeling)

About World Labs:

We build foundational world models that can perceive, generate, reason, and interact with the 3D world — unlocking AI's full potential through spatial intelligence by transforming seeing into doing, perceiving into reasoning, and imagining into creating.

We believe spatial intelligence will unlock new forms of storytelling, creativity, design, simulation, and immersive experiences across both virtual and physical worlds.

We bring together a world-class team, united by a shared curiosity, passion, and deep backgrounds in technology — from AI research to systems engineering to product design — creating a tight feedback loop between our cutting-edge research and products that empower our users.

Role Overview

We are looking for a talented Research Scientist with a strong background in generative modeling, particularly diffusion models, to join our modeling team. This role is ideal for candidates with deep expertise in diffusion models applied to images, videos, or 3D assets and scenes.

While not required, experience in one or more of the following areas is a strong plus:

• Data curation for pretraining or post-training
• Tokenizers and VAEs for image, video, or 3D data
• 3D vision

You will collaborate closely with researchers, engineers, and product teams to bring advanced 3D modeling and machine learning techniques into real-world applications, ensuring that our technology remains at the forefront of visual innovation. This role involves significant hands-on research and engineering work, driving projects from conceptualization through to production deployment.

What You Will Do:

• Design, implement, and train large-scale diffusion models for generating 3D worlds
• Develop and experiment with large-scale diffusion models to add novel control signals, adapt to target aesthetic preferences, or distill for efficient inference
• Collaborate closely with research and product teams to understand and translate product requirements into effective technical roadmaps
• Contribute hands-on to all stages of model development including data curation, experimentation, evaluation, and deployment
• Continuously explore and integrate cutting-edge research in diffusion and generative AI more broadly
• Act as a key technical resource within the team, mentoring colleagues, and driving best practices in generative modeling and ML engineering

Key Qualifications:

• 3+ years of experience in generative modeling or applied ML roles, ideally at a startup or other fast-paced research environment
• Extensive experience with machine learning frameworks such as PyTorch or TensorFlow, especially in the context of diffusion models and other generative models
• Deep expertise in at least one area of generative modeling: pre-training, post-training, diffusion distillation, fine-tuning with new conditioning signals, etc for diffusion models
• Strong history of publications or open-source contributions involving large-scale diffusion models
• Strong coding proficiency in Python and experience with GPU-accelerated computing
• Ability to engage effectively with researchers and cross-functional teams, clearly translating complex technical ideas into actionable tasks and outcomes
• Comfortable operating within a dynamic startup environment with high levels of ambiguity, ownership, and innovation.

Preferred Qualifications:

• Contributions to open-source projects in the fields of computer vision, graphics, or ML
• Familiarity with large-scale training infrastructure (e.g., multi-node GPU clusters, distributed training environments)
• Experience integrating machine learning models into production environments
• Led or been involved with the development or training of large-scale, state-of-the-art generative models

Who You Are:

• Fearless Innovator: We need people who thrive on challenges and aren't afraid to tackle the impossible
• Resilient Builder: Impacting Large World Models isn't a sprint; it's a marathon with hurdles. We're looking for builders who can weather the storms of groundbreaking research and come out stronger
• Mission-Driven Mindset: Everything we do is in service of creating the best spatially intelligent AI systems, and using them to empower people
• Collaborative Spirit: We're building something bigger than any one person. We need team players who can harness the power of collective intelligence

We're hiring the brightest minds from around the globe to bring diverse perspectives to our cutting-edge work. If you're ready to work on technology that will reshape how machines perceive and interact with the world - then World Labs is your launchpad.

Join us, and let's make history together.

Equal Opportunity & Pay Transparency

World Labs is an equal opportunity employer. We do not discriminate on the basis of race, color, religion, sex, sexual orientation, gender identity, national origin, age, disability, genetic information, veteran status, or any other characteristic protected under applicable law. We welcome all qualified applicants and are committed to providing reasonable accommodations throughout the hiring process upon request.

In accordance with California law, we disclose the following:

Pay Range

$250,000 - $325,000 base salary (good-faith estimate for San Francisco Bay Area upon hire; actual offer based on experience, skills, and qualifications)

Total Compensation

Base salary plus equity awards and annual performance bonus