# Research Scientist (Generative Modeling)

**Company:** [World Labs](https://hotfix.jobs/companies/world-labs)
**Location:** San Francisco, CA
**Role:** AI Research
**Experience:** 3+ years
**Skills:** Diffusion Models, PyTorch, TensorFlow, Python, Generative Modeling, 3D Vision, Gpu Computing, Vaes, Tokenizers, Distributed Training
**Posted:** 2026-04-30

> Develops and trains large-scale diffusion models for generating 3D worlds, incorporating novel control signals and integrating cutting-edge generative AI research into production. Requires 3+ years in generative modeling, PyTorch/TensorFlow expertise, and strong publication record.

## Job Description

## 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

While not required, experience in one or more of the following areas is a strong plus:
- Large-scale model training
- Data curation for pretraining or post-training
- Tokenizers and VAEs for image, video, or 3D data
- Long-context architectures
- 3D vision

## Similar roles

- [ML Scientist](https://hotfix.jobs/jobs/401a3fda-7a0b-4af9-aa9c-134e4ae4b19c) - Sesame - San Francisco, CA - $190k – $320k/yr
- [Research Scientist](https://hotfix.jobs/jobs/34963160-7769-44e3-8e41-872372a811ed) - OpenRouter - Remote
- [Research Engineer, Benchmarks](https://hotfix.jobs/jobs/71636332-204e-4abc-b0b5-c2b5c2ca3726) - hud - San Francisco, CA
- [Research Scientist](https://hotfix.jobs/jobs/5c6acbfc-713c-4c38-bb2c-7a8d9dbc74d1) - Spotify - Remote
- [Research Scientist, Life Sciences](https://hotfix.jobs/jobs/3612b705-0fb1-4f66-a4a1-0b28ba382b5c) - Anthropic - San Francisco, CA - $300k – $320k/yr

**Apply:** https://hotfix.jobs/jobs/f1a9eb1c-7358-4dee-bfe3-d00ba8dffa04
**Canonical:** https://hotfix.jobs/jobs/f1a9eb1c-7358-4dee-bfe3-d00ba8dffa04