Applied ML Research – Image/Video Diffusion
Job Description
- Train diffusion models for image and video generation
- Train controllability modules like IPAdapters or ControlNets
- Develop novel research techniques and put them into production
- Conduct large-scale experiments on HPC clusters
- Experience training image models at scale
- Proven track record with image/video models
- Strong deep learning framework background
- Expertise in distributed training paradigms
- Ability to iterate rapidly
- Propose creative research directions
- Sponsorship for International candidates (STEM OPT, OPT, H1B, O1, E3)
- Work with a world-class team developing AI tooling
- Significant impact on Krea's market presence and growth
- Competitive compensation (75th percentile of market rates)
- Significant equity upside
About Krea
At Krea, we're dedicated to making AI intuitive and controllable for creatives. Our mission is to build tools that empower human creativity, not replace it. We believe AI is a new medium that allows us to express ourselves through various formats—text, images, video, sound, and even 3D. We're building better, smarter, and more controllable tools to harness this medium.
We’re backed by Bain Capital Ventures, A16Z, Abstract Ventures, Pebblebed and many others. If you're passionate about pushing the boundaries of AI and empowering human creativity, we'd love to hear from you.
We're looking for an experienced Researcher with engineering skills who can work on large-scale image and video models training experiments, with experience training image models at scale on past and relevant publications.
Our culture:
We work full-time and in-person at our North Point office in San Francisco.
We believe that demonstrated interest in the creative space is key: our team includes musicians, designers, visual artists and more.
What you'll do:
Train diffusion models for image and video generation.
Train controllability modules such as IPAdapters or ControlNets.
Develop novel research techniques and put them into production.
Conducting large-scale experiments on high-performance computing clusters, optimizing data pipelines for massive image datasets
Example experience and skills we’re looking for
Proven track record in working with image or video models at scale (publications or open-source contributions a plus)
Strong background in deep learning frameworks and distributed training paradigms.
Ability to iterate rapidly, and propose creative research directions
What we offer:
Openness to sponsoring International candidates (e.g STEM OPT, OPT, H1B, O1, E3)
Work alongside a world class developing the future of AI tooling
Significant impact on Krea’s market presence and growth
Competitive compensation (75% percentile of market rates) with significant equity upside
