Machine Learning Engineer

Who you are

You have a proven track record in working with image or video models at scale — whether through publications or open-source contributions, your experience speaks volumes about your capabilities. You possess a strong background in deep learning frameworks and distributed training paradigms, allowing you to tackle complex challenges effectively. Your ability to iterate rapidly and propose creative research directions sets you apart in the field of machine learning.

You thrive in environments that require large-scale experiments on high-performance computing clusters — optimizing data pipelines for massive image datasets is second nature to you. Your passion for AI and its applications in creative tools drives your work, and you are eager to contribute to projects that empower human creativity.

What you'll do

As a Machine Learning Engineer at Krea, you will train foundation diffusion models for image and video generation, pushing the boundaries of what AI can achieve in creative contexts. You will also work on developing controllability modules such as IPAdapters or ControlNets, ensuring that the tools we create are intuitive and controllable for users. Your role will involve conducting large-scale experiments, optimizing data pipelines, and collaborating with a team dedicated to building next-generation AI creative tools.

You will have the opportunity to develop novel research techniques and put them into production, contributing to the evolution of AI as a medium for creative expression. Your work will directly impact the tools that creatives use, making AI a more accessible and powerful resource for various formats including text, images, video, sound, and 3D.

What we offer

At Krea, we are backed by world-class Silicon Valley investors and have raised over $83M to support our mission. You will be part of a dynamic team that values creativity and innovation, working in an environment that encourages experimentation and growth. We believe in empowering our employees and providing them with the resources they need to succeed in their roles. Join us in shaping the future of AI and creativity.