Lead Machine Learning Engineer (Personalization & AI Models) Full-time

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.

The remote option applies only to the Candidates who will be working from any location in Ukraine.

#LI-DNI

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

• 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)
• Consider relocation options in over 30 countries worldwide
• Participate in volunteer, charity programs and communities (both technical and interest-based)
• 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)
• 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

EPAM strives to provide its global team of over 52,800+ 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.