Drimva

Krea

ML Engineer - Personalization & Recommendation Systems

San Francisco · On-site · Full-time · Posted Dec 17, 2025

Visa sponsorship
Open to sponsoring international visas where possible (e.g., STEM OPT, OPT, H-1B, O-1, E-3).

About the job

Architect and build personalization and recommendation systems that model user taste, curate creative feeds, and adapt generative models to individual aesthetics.

What you'll do

  • Build and own the personalization and recommendation stack from research through production.
  • Develop user-preference algorithms and curated feeds balancing exploration, personalization, and aesthetic coherence.
  • Collaborate with model, research, product, and design teams on personalization for generative image and video tools.

What they ask for

  • Strong experience building recommendation systems or personalized feeds from scratch.
  • Proven ability to design and ship curated content experiences.
  • Experience with media-based personalization; image or video preferred, with other modalities also welcome.
  • Solid machine-learning foundations, strong Python skills, and experience with PyTorch or JAX.

Nice to have

  • Experience with large-scale data systems or production ML infrastructure.
  • Familiarity with diffusion models, generative image systems, embeddings, similarity search, ranking, or aesthetic evaluation.

Tools

Python, PyTorch, JAX

AI focus

Personalization for generative image and video tools, including adapting models to individual styles and contributing to personalized image-generation research.

Model-written summary · the original posting is authoritative

First seen Oct 10, 2026 · Last verified Nov 5, 2026 (UTC)