Machine Learning Engineer

About KreaAt Krea, we are building next-generation AI creative tools.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 recently took this a step forward with the launch of Krea 2, our first foundation model, built completely from scratch for aesthetic diversity and stylistic control.We've raised over $83M and are backed by world-class investors such as a16z, Bain Capital, and Abstract. We work full-time and in-person at our waterfront office in San Francisco. We care about creativity: our team includes musicians, designers, visual artists, and engineers.This roleWe're looking for a machine learning engineers who can work on large-scale image and video models training experiments.What you'll doTrain foundation diffusion models for image and video generationTrain controllability modules such as IPAdapters or ControlNetsDevelop novel research techniques and put them into productionConducting large-scale experiments on high-performance computing clusters, optimizing data pipelines for massive image datasetsWhat we're looking forProven 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 paradigmsAbility to iterate rapidly, and propose creative research directionsWhat we offerTeam: Work alongside a world-class team building the future of AI creative toolingImpact: Significant scope and company-wide impactCompetitive compensation: generous salary & equity packagesHealth & wellness: 100% health & 99% dental/vision insurance premiums covered for employees, health FSA accounts, & long-term disability coverageTime off: Flexible PTO policyFinancial planning: 401k with a 4% company-sponsored matchMeals in the office: breakfast, lunch, dinner - you name it, we'll cover itTransit: Ubers covered to & from the officeSponsorship: We're open to sponsoring international visas where we can (e.g., STEM OPT, OPT, H-1B, O-1, E-3). And more!Please note the above benefits & perks are for full-time employees