ML Engineer - Personalization & Recommendation Systems

The role

You will architect and build Krea's personalization and recommendation systems from the ground up. You will design algorithms to model user preference and work with the model and research team to co-design personalization mechanisms.

What you will do

• Architect and build Krea's personalization and recommendation stack from the ground up
• Design algorithms to model user preference and taste
• Build high-quality, curated feeds that balance exploration, personalization, and aesthetic coherence
• Work with the model and research team to co-design personalization mechanisms
• Contribute to personalized image generation research
• Collaborate with product, design, and research to define personalization
• Take systems from research and prototyping through production, iteration, and continuous improvement

Required qualifications

• Strong experience building recommendation systems or personalized feeds from scratch
• Proven ability to design and ship high-quality curated content experiences
• Experience working with media-based personalization
• Solid foundations in machine learning, representation learning, and modern deep learning techniques
• Strong Python skills
• Experience with ML frameworks such as PyTorch or JAX
• Ability to operate independently, make architectural decisions, and own complex systems end to end

Nice to have

• Experience with large-scale data systems and production ML infrastructure
• Prior work on or familiarity with diffusion models or generative image systems
• Background in embeddings, similarity search, ranking, or aesthetic evaluation
• Interest in creative tools, art, design, or generative media
• Experience with image, video preferred; music or other modalities also welcome

Benefits

• Generous salary & equity packages
• 100% health insurance premiums covered for employees
• 99% dental/vision insurance premiums covered for employees
• Health FSA accounts
• Long-term disability coverage
• Flexible PTO policy
• 401k with a 4% company-sponsored match
• Breakfast, lunch, dinner covered in the office
• Ubers covered to & from the office

Before you apply

• We're open to sponsoring international visas where we can (e.g., STEM OPT, OPT, H-1B, O-1, E-3)

About the company

Krea is building next-generation AI creative tools to empower human creativity. They have raised over $83M and are backed by investors such as a16z, Bain Capital, and Abstract.