We are seeking a Lead Machine Learning Engineer (Personalization & AI Models) to pioneer the building and optimization of user segmentation, recommendation, and embedding models within our expansive personalization system for a Mobile App.
The focus areas will include multi-vector representations, real-time model inference, and the integration of personalization workflows with cutting-edge technologies including AWS Personalize SDK, PGVector, and RudderStack.
Responsibilities
- Develop and optimize embedding models using sentence-transformer models for user profiles and personalization
- Implement KNN-based recommendation systems for real-time content scoring and ranking
- Utilize AWS Personalize SDK to train and deploy machine learning models that dynamically adapt to user behavior
- Integrate embedding models with Databricks, RudderStack, and AWS services to ensure real-time profile updates
- Fine-tune ML models aimed at boosting revenue predictions, enhancing user engagement, and refining audience segmentation
- Optimize PGVector and Redis for efficient vector-based lookups and caching
- Collaborate with data engineers to devise ML pipelines for robust training, validation, and deployment processes
Requirements
- Strong experience in Machine Learning, Deep Learning, and AI-driven personalization
- Proficiency in Python, PyTorch, and TensorFlow
- Expertise in vector-based search and recommendation systems, including knowledge of KNN, PGVector, Redis
- Hands-on experience with AWS Personalize SDK and SageMaker or similar ML training pipelines
- Familiarity with Databricks, Delta Lake, and Apache Spark for large-scale model training and deployment
- Strong understanding of real-time personalization, A/B testing, and user segmentation models
- Capability to work with event-driven architectures and implement real-time feature engineering
- Fluent English communication skills at a B2+ level
Nice to have
- Knowledge of ML model monitoring, MLOps, and automation of ML pipelines through CI/CD
- Experience with Graph Neural Networks (GNNs) for analyzing user similarity and clustering
- Familiarity with deployment of real-time analytics tools and dashboarding, such as Looker, Tableau, or Snowflake
Benefits
With us you can:
- Work on a flexible schedule remotely or from any of our comfortable offices or coworking spaces in Ukraine
- Receive the necessary equipment to perform your work tasks
- Change projects and technology stacks within EPAM
- Gain experience in various business domains (Insurance, E-commerce, Healthcare, Finance, Travelling, Media, Artificial Intelligence, and more)
- Relocation opportunities may be available for eligible candidates, depending on the role and openings at other EPAM locations
- Participate in volunteer, charity programs and communities (both technical and interest-based)
We focus on your professional growth:
- You can plan your individual career path together with your manager
- Receive regular feedback from colleagues
- Improve your English for free with certified teachers (Speaking Clubs, client interview preparation courses, etc.)
- Get the opportunity to undergo free training and certification in AWS, GCP, or Azure Clouds
- Use the internal E-learn training program (18,200+ specialized training and mentoring programs)
- Access corporate accounts on LinkedIn Learning, Get Abstract and other partner resources
- Study at EPAM Solution Architecture School with the instructors who are practicing architects
- Develop as a leader, join Delivery Management, Resource Management, Leadership Essentials school and more
- Participate in internal communities (500+ meetups, technical discussions, brainstorming sessions, online events and conferences annually)
What we offer:
- Vacation and sick leave (including a sick leave without a medical certificate)
- A wide range of Voluntary Medical Insurance programs providing both medical treatment and various preventive options (including sports activities)
- Medical insurance for family members at corporate rates
- Company support during significant life events (childbirth or adoption, marriage, etc.)
- Support for psychological comfort: discounts on services from mental health specialists or coaches, thematic training
- E-kids program - a free programming language training program for EPAMers' children