Machine Learning Engineer

About The Company

Krea is a pioneering technology company dedicated to advancing the frontiers of artificial intelligence and machine learning. Our mission is to develop innovative solutions that empower creators, artists, and developers to harness the full potential of AI-driven tools for image and video generation. With a focus on cutting-edge research and development, Krea strives to push the boundaries of what is possible in the realm of AI-powered creative applications. Our collaborative and inclusive culture fosters continuous learning, experimentation, and growth, making us a leader in the AI industry.

About The Role

We are seeking a highly skilled Machine Learning Engineer to join our dynamic team at Krea. In this role, you will be at the forefront of developing large-scale image and video models, conducting extensive training experiments, and optimizing cutting-edge diffusion models for creative content generation. Your expertise will contribute to building scalable, efficient, and innovative AI models that redefine the possibilities of digital art and media. This position offers an exciting opportunity to work on high-impact projects with a talented team dedicated to shaping the future of AI-powered creative tools.

Qualifications

Searching for Machine Learning & AI roles that provide visa sponsorship? Connect with international employers through Machine Learning & AI Jobs with Visa Sponsorship opportunities actively seeking talented professionals.

The ideal candidate will possess a proven track record in working with image or video models at scale, demonstrated through publications, open-source contributions, or industry experience. A strong background in deep learning frameworks such as PyTorch or TensorFlow, coupled with expertise in distributed training paradigms, is essential. Candidates should demonstrate the ability to iterate rapidly, propose innovative research directions, and translate them into production-ready solutions. A solid understanding of diffusion models, controllability modules like IPAdapters or ControlNets, and experience with high-performance computing clusters are highly desirable. Excellent problem-solving skills, creativity, and a passion for advancing AI research are key attributes we seek.

Responsibilities

• Train foundation diffusion models for image and video generation, ensuring high quality and scalability.
• Develop and implement controllability modules such as IPAdapters or ControlNets to enhance model flexibility and user control.
• Conduct large-scale experiments on high-performance computing clusters, optimizing data pipelines for massive image and video datasets.

Explore our comprehensive directory of visa sponsorship jobs from employers worldwide who are ready to sponsor talented international professionals.

• Design, test, and deploy novel research techniques, translating cutting-edge research into production environments.
• Collaborate with cross-functional teams to integrate AI models into product workflows and creative tools.
• Monitor and analyze model performance, troubleshooting issues, and iterating to improve results.
• Stay current with the latest research developments in AI and machine learning, contributing to the continuous innovation at Krea.
  
  

Interested in opportunities specifically in United State? Discover our dedicated Visa Sponsorship Jobs in United State page featuring roles from top employers in this location.

Benefits

At Krea, we believe in supporting our employees both professionally and personally. We offer a comprehensive benefits package that includes a competitive salary and equity options, designed to reward your contributions and growth. Our health benefits cover 100% of health insurance premiums and 99% of dental and vision plans, ensuring you and your family are well-protected. We also provide health FSA accounts and long-term disability coverage. Our flexible PTO policy allows you to take time off as needed to recharge and maintain work-life balance. Additionally, we offer a 401(k) plan with a 4% company-sponsored match to support your financial future. Our amenities include catered breakfast, lunch, and dinner, and transportation support with Uber coverage for commuting. We are open to sponsoring international visas, including STEM OPT, OPT, H-1B, O-1, and E-3, making this opportunity accessible to talented professionals worldwide.

Equal Opportunity

Krea is an equal opportunity employer committed to fostering a diverse and inclusive workplace. We do not discriminate based on race, ethnicity, gender, sexual orientation, age, disability, or any other protected characteristic. We believe that diversity drives innovation and creativity, and we are dedicated to providing an environment where all employees can thrive and contribute to our shared success.