Krea AI
ML Engineer - Personalization & Recommendation Systems
San Francisco · Full-time
Sponsorship not specifiedDetected 304 days ago
PythonExpressAlgorithmsMachine LearningDeep LearningPyTorchMLOpsResearch
Stay score
odds of building a lasting career here
34Risky
Cap-exempt (no lottery)0
Sponsors this role68
Entry-level history0
PERM / green-card track0
Lottery odds40
Fits your clock70
Thin sponsorship signal and lottery-bound. A low-probability bet with your clock running. Prioritize cap-exempt roles and proven entry-level sponsors first.
Lottery odds assume a STEM candidate.
Personalize to your clock →Community outcomes
No reports yet — be the first to help the next applicant.
About the role
- We believe AI is a new medium that allows us to express ourselves through various formats - text, images, video, sound, and even 3D.
- We've raised over $83M and are backed by world-class investors such as a16z, Bain Capital, and Abstract.
- We work full-time and in-person at our waterfront office in San Francisco.
Responsibilities
- 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 our model and research team to co-design personalization mechanisms that influence how our generative models learn, adapt, and express style
- Collaborate closely with product, design, and research to define what "good personalization" means in a creative context
- Strong experience building recommendation systems or personalized feeds from scratch
- Proven ability to design and ship high-quality curated content experiences
- Team: Work alongside a world-class team building the future of AI creative tooling
Requirements
- Strong Python skills and experience with ML frameworks such as PyTorch or JAX
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
- Experience working with media-based personalization (image, video preferred
- music or other modalities also welcome)
Compensation
- generous salary & equity packages
Benefits
- Competitive compensation: generous salary & equity packages
- Health & wellness: 100% health & 99% dental/vision insurance premiums covered for employees, health FSA accounts, & long-term disability coverage
- Time off: Flexible PTO policy
- Please note the above benefits & perks are for full-time employees
- Solid foundations in machine learning, representation learning, and modern deep learning techniques
Visa & Work Authorization
- We're open to sponsoring international visas where we can (e.g., STEM OPT, OPT, H-1B, O-1, E-3).
Apply directly at Krea AI →Create a free account for alerts like thisView Krea AI immigration profile
This listing is sourced directly from Krea AI's careers page and normalized into a canonical job model.