ML Engineer - Personalization & Recommendation Systems

About the role

We are building next-generation AI creative tools at Krea. Our mission is to empower human creativity through intuitive and controllable AI, focusing on tools for text, images, video, sound, and 3D expression.

Responsibilities

• Arcitect and build Krea’s personalization and recommendation stack from the ground up, owning the technical direction end to end.
• Design algorithms to model user preference and taste, enabling Krea’s models to adapt to individual styles and aesthetics.
• Build high-quality, curated feeds that balance exploration, personalization, and aesthetic coherence.
• Work directly with the model and research team to co-design personalization mechanisms that influence how our generative models learn, adapt, and express style.
• Contribute to personalized image generation research, with a focus on style, taste, and subjective quality.
• Collaborate closely with product, design, and research to define what “good personalization” means in a creative context.
• Take systems from research and prototyping through production, iteration, and continuous improvement.

Requirements

• 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 (image, video preferred; music or other modalities also welcome).
• Solid foundations in machine learning, representation learning, and modern deep learning techniques.
• Strong Python skills and experience with ML frameworks such as PyTorch or JAX.
• Ability to operate independently, make architectural decisions, and own complex systems end to end.

Qualifications

• Bonus points for 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.

Skills

• 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 (image, video preferred; music or other modalities also welcome).
• Solid foundations in machine learning, representation learning, and modern deep learning techniques.
• Strong Python skills and experience with ML frameworks such as PyTorch or JAX.
• Ability to operate independently, make architectural decisions, and own complex systems end to end.

Benefits

• Openness to sponsoring international candidates (e.g., STEM OPT, OPT, H-1B, O-1, E-3).
• Work alongside a world-class team building the future of AI creative tooling.
• Significant scope and company-wide impact.
• Competitive compensation (75th percentile of market) with meaningful equity.

Pay

Competitive compensation (75th percentile of market) with meaningful equity.

Schedule

Not specified.