Machine Learning Engineer
Krea is an active generative-AI creative platform for generating, editing, and enhancing images, video, and 3D content.
San Francisco, United States
Funding history
About Krea
Krea provides browser-based AI creative tools and a developer API. Its offerings include image and video generation, enhancement and editing tools, 3D generation, and Krea 2, its in-house image foundation model.
Skills
Candidate Availability
Required and preferred rules are kept separate and reflect the wording in the original posting.
About the Role
You will train foundation diffusion models for image and video generation, develop controllability modules, and turn research techniques into production systems. You will conduct large-scale experiments on high-performance computing clusters and optimize data pipelines for image datasets.
Requirements
- Experience with image or video models at scale
- Deep learning framework expertise
- Distributed training expertise
- Research direction development
Responsibilities
- Train foundation diffusion models for image and video generation
- Train controllability modules such as IPAdapters and ControlNets
- Develop novel research techniques and put them into production
- Conduct large-scale experiments on high-performance computing clusters
- Optimize data pipelines for massive image datasets
Benefits
- Health insurance with 100% employee premium coverage
- Dental and vision insurance with 99% employee premium coverage
- Health FSA accounts
- Long-term disability coverage
- Flexible PTO
- 401k with a 4% company-sponsored match
- Office meals
- Uber transit to and from the office
- International visa sponsorship where possible