The company is seeking a Research Scientist specializing in generative modeling and diffusion models to bring advanced 3D modeling and machine learning techniques into real-world applications, ensuring their technology remains at the forefront of visual innovation.
Requirements
- pre-training or post-training of large-scale diffusion models for images, videos, or 3D assets or scenes
- machine learning frameworks such as PyTorch or TensorFlow, especially in the context of diffusion models and other generative models
- pre-training, post-training, diffusion distillation, etc for diffusion models
- Strong coding proficiency in Python and experience with GPU-accelerated computing.
- 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
Responsibilities
- Design, implement, and train large-scale diffusion models for generating 3D worlds
- Develop and experiment with post-training for large-scale diffusion models to add novel control signals, adapt to target aesthetic preferences, or distill for efficient inference
- 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
Other
- Collaborate closely with researchers, engineers, and product teams
- Collaborate closely with research and product teams to understand and translate product requirements into effective technical roadmaps.
- 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.
- 3+ years of experience in generative modeling or applied ML roles, ideally at a startup or other fast-paced research environment
