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. ResponsibilitiesDevelop and optimize embedding models using sentence-transformer models for user profiles and personalizationImplement KNN-based recommendation systems for real-time content scoring and rankingUtilize AWS Personalize SDK to train and deploy machine learning models that dynamically adapt to user behaviorIntegrate embedding models with Databricks, RudderStack, and AWS services to ensure real-time profile updatesFine-tune ML models aimed at boosting revenue predictions, enhancing user engagement, and refining audience segmentationOptimize PGVector and Redis for efficient vector-based lookups and cachingCollaborate with data engineers to devise ML pipelines for robust training, validation, and deployment processes RequirementsStrong experience in Machine Learning, Deep Learning, and AI-driven personalizationProficiency in Python, PyTorch, and TensorFlowExpertise in vector-based search and recommendation systems, including knowledge of KNN, PGVector, RedisHands-on experience with AWS Personalize SDK and SageMaker or similar ML training pipelinesFamiliarity with Databricks, Delta Lake, and Apache Spark for large-scale model training and deploymentStrong understanding of real-time personalization, A/B testing, and user segmentation modelsCapability to work with event-driven architectures and implement real-time feature engineeringFluent English communication skills at a B2+ level Nice to haveKnowledge of ML model monitoring, MLOps, and automation of ML pipelines through CI/CDExperience with Graph Neural Networks (GNNs) for analyzing user similarity and clusteringFamiliarity with deployment of real-time analytics tools and dashboarding, such as Looker, Tableau, or Snowflake We offerWith us you can:Work on a flexible schedule remotely or from any of our comfortable offices or coworking spaces in UkraineReceive the necessary equipment to perform your work tasksChange projects and technology stacks within EPAMGain 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 locationsParticipate 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 managerReceive regular feedback from colleaguesImprove 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 CloudsUse the internal E-learn training program (18,200+ specialized training and mentoring programs)Access corporate accounts on LinkedIn Learning, Get Abstract and other partner resourcesStudy at EPAM Solution Architecture School with the instructors who are practicing architectsDevelop as a leader, join Delivery Management, Resource Management, Leadership Essentials school and moreParticipate 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 ratesCompany support during significant life events (childbirth or adoption, marriage, etc.)Support for psychological comfort: discounts on services from mental health specialists or coaches, thematic trainingE-kids program - a free programming language training program for EPAMers' children Kindly note that this role supports remote work, but only from within Ukraine. 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 61,700 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.