Machine Learning Engineer

Krea.ai, Inc

$120K — $180K *
Information Technology
5 - 7 years of experience
Job Overview by Ladders

Qualifications

  • 5-7 years of experience in machine learning or computer vision, especially with image/video models.
  • Strong proficiency in deep learning frameworks like TensorFlow or PyTorch.
  • Experience with distributed training across large datasets in high-performance computing environments.
  • Demonstrated ability to produce results with publications or open-source projects related to AI creativity tools.
  • Creative problem-solving skills and willingness to iterate on research ideas.

Responsibilities

  • Train advanced diffusion models for generating images and videos.
  • Develop and implement controllability models like IPAdapters or ControlNets.
  • Execute large-scale experiments leveraging high-performance computing resources.
  • Optimize data pipelines to manage massive image datasets efficiently.
  • Translate novel research techniques into practical applications for creative tools.

Benefits

  • Collaborative work environment with a focus on innovation and creativity.
  • Access to cutting-edge technology and resources for ML experimentation.
  • Opportunity to shape the future of AI in creative domains through impactful projects.
  • Supportive team culture with a focus on professional development and growth.
  • Backed by leading investors and influencers in the AI and tech industry.
Full Job Description
This job

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

Some stuff you can 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


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

Similar Jobs

More Jobs at Krea.ai, Inc

  • Account Executive
    $120K — $150K *
    San Francisco, CA 94112 (San Francisco County)
    Business Services
    In-Person
  • ML Engineer - Personalization & Recommendation Systems
    $120K — $180K *
    San Francisco, CA 94112 (San Francisco County)
    Consumer Technology
    In-Person
  • Fullstack Engineer
    $120K — $160K *
    San Francisco, CA 94112 (San Francisco County)
    Consumer Technology
    In-Person
  • Machine Learning Engineer
    $120K — $180K *
    San Francisco, CA 94112 (San Francisco County)
    Information Technology
    In-Person
  • Growth Marketer
    $120K — $150K *
    San Francisco, CA 94112 (San Francisco County)
    Business Services
    In-Person

More Information Technology Jobs

Find similar Machine Learning Engineer jobs: