Krea AI
ML Engineer - Personalization & Recommendation Systems
San Francisco
Sponsorship not specifiedDetected 218 days ago
PythonExpressAlgorithmsMachine LearningDeep LearningPyTorchMLOpsResearch
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.
- 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.
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
- 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.
- Experience with large-scale data systems and production ML infrastructure
Nice to have
- Experience working with media-based personalization (image, video preferred
- music or other modalities also welcome)
Compensation
- Competitive compensation (75th percentile of market) with meaningful equity.
Benefits
- Competitive compensation (75th percentile of market) with meaningful equity.
- Solid foundations in machine learning, representation learning, and modern deep learning techniques
Company info
- What we're looking for
Visa & Work Authorization
- Openness to sponsoring international candidates (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.