Lead Machine Learning Engineer (Personalization & AI Models)

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.

Kindly note that this role supports remote work, but only from within Ukraine.

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

We offer

• 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

Kindly be advised that the set of benefits, including learning, certification, and other opportunities, may vary depending on the role you apply for. Our recruiter will be able to share more details about the specific opportunity during your general interview.

EPAM strives to provide its global team of over 62,350 professionals in more than 55 countries with opportunities for professional growth from day one of collaboration. Our colleagues are the source of EPAM's success, so we value cooperation, strive to always understand our clients' business and aim for the highest quality standards. No matter where you are, you will join a dedicated, diverse community that will help you realize your potential to the fullest.