Machine Learning Engineer

Krea

  • San Francisco, California
  • 30+ days ago

    Highlights

    Conducting large-scale experiments on high-performance computing clusters, optimizing data pipelines for massive image datasets. Proven track record in working with image or video models at scale (publications or open-source contributions a plus).

    Numbers & Facts

    LocationSan Francisco, California

    Description

    About Krea

    At 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 role

    We're looking for a machine learning engineers who can work on large-scale image and video models training experiments.

    What you'll do

    • Train foundation 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

    What 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

    • Team: Work alongside a world-class team building the future of AI creative tooling

    • Impact: Significant scope and company-wide impact

    • Competitive compensation: generous salary & equity packages

    • Health & wellness: 100% health & 99% dental/vision insurance premiums covered for employees, health FSA accounts, & long-term disability coverage

    • Time off: Flexible PTO policy

    • Financial planning: 401k with a 4% company-sponsored match

    • Meals in the office: breakfast, lunch, dinner - you name it, we'll cover it

    • Transit: Ubers covered to & from the office

    • Sponsorship: 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

    Similar Jobs

    See more jobs