ML Engineer - Personalization & Recommendation Systems

krea.ai; San Francisco (CA); On-site; USD 140,000 - 200,000; Full time

Benefits offered by this job

Competitive compensation

Openness to sponsoring international candidates

Impact in a growing team

Job summary

A leading AI creative tooling company in California is seeking an ML Engineer to create and manage their personalization and recommendation systems. The ideal candidate will have strong experience in building personalized feeds, proficiency in Python, and a solid foundation in machine learning. Responsibilities include shaping user experience through tailored content, collaborating with multiple teams, and contributing to generative media research. The position offers competitive compensation and opportunities for international sponsorship.

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 (image, video preferred).
• Solid foundations in machine learning, representation learning, and modern deep learning techniques.
• Strong Python skills and experience with ML frameworks.

Responsibilities

• Architect and build Krea’s personalization and recommendation systems.
• Design algorithms to model user preferences.
• Build high-quality, curated feeds.
• Work with model and research teams on personalization mechanisms.
• Contribute to personalized image generation research.
• Collaborate with product, design, and research.

Skills

Building recommendation systems

Designing curated content experiences

Machine learning techniques

Python programming

Working with ML frameworks (e.g., PyTorch, JAX)

Operating independently

Job description

Join to apply for the ML Engineer - Personalization & Recommendation Systems role at krea.ai.

About Krea

At Krea, we are building next-generation AI creative tools. We are 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.

This Job

We’re looking for an ML Engineer to architect and build Krea’s personalization and recommendation systems from scratch. You’ll have full ownership over how we understand user taste, curate content, and adapt generative models to individual aesthetics. This is a role at the intersection of recommendation systems, representation learning, and generative image and video tools.

What You’ll Do

• Architect 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.

What We’re Looking For

• 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.

Bonus points

• 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.

What we offer

• 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.

Seniority level

Mid-Senior level

Employment type

Full-time

Job function

Engineering and Information Technology

Industries

Artists and Writers