Lead AI Solution Engineer with Databricks, ML

We are looking for a Lead AI Solution Engineer who brings strong expertise in Databricks and Machine Learning to architect, launch, and manage production-level ML systems. The ideal candidate works hands-on as a Data Engineer or Machine Learning Engineer, has demonstrated success delivering both real-time and batch ML solutions, possesses solid skills in building streaming pipelines, and brings experience maintaining the scalability and reliability of ML systems once they are live in production.

Responsibilities Design and optimize feature engineering pipelines for both batch and real-time workloads Build, deploy, and maintain scalable model training and inference pipelines Process structured and unstructured data at scale Develop streaming solutions using Spark Streaming Enable real-time feature generation and model serving Ensure reliability, scalability, and performance of ML solutions in production Support batch pipelines for model training and retraining Operate and maintain production ML systems to ensure consistent performance Requirements 7+ years of experience building, deploying, and operating offline and online feature engineering pipelines Proficiency in Databricks, Spark/PySpark, Python, and SQL Expertise in model training and real-time inferencing pipelines Experience processing large-scale unstructured data Knowledge of Spark Streaming Experience with real-time feature generation and low-latency model inference Proven track record of running production-grade ML systems English proficiency at B2 level or higher Nice to have Exposure to MLflow and established MLOps workflows Hands-on background with AWS Prior work in Customer Analytics, Recommendation Systems, or Retail industries Armenia (Prod) We connect like-minded people Delivering innovative solutions to industry leaders, making a global impact Enjoyable working environment, whether it is the vibrant office or the comfort of your home Opportunity to work abroad for up to two months per year Relocation opportunities within our offices in 55+ countries Corporate and social events We invest in your growth Leadership development, career advising, soft skills and well-being programs Certifications, including GCP, Azure and AWS Unlimited access to EPAM's internal learning database Free English classes with certified teachers We cover it all Participation in the Employee Stock Purchase Plan Monetary bonuses for engaging in the referral program Comprehensive medical & family care package Four trust days per year for personal needs Discounts for fitness clubs Benefits package (hotels, restaurants, stores and services) Georgia (Production) We connect like-minded people Delivering innovative solutions to industry leaders, making a global impact Enjoyable working environment, whether it is the vibrant office or the comfort of your own home Opportunity to work abroad for up to two months per year Relocation opportunities within our offices in 55+ countries Corporate and social events We invest in your growth Leadership development, career advising, soft skills and well-being programs Certifications, including GCP, Azure and AWS Unlimited access to EPAM's internal learning database Free English classes with certified teachers We cover it all Participation in the Employee Stock Purchase Plan Monetary bonuses for engaging in the referral program Comprehensive medical & family care package Five trust days per year (sick leave without a medical certificate) Benefits package (sports activities, a variety of stores and services) Kazakhstan (Prod) We connect like-minded people: Delivering innovative solutions to industry leaders, making a global impact Enjoyable working environment, whether it is the vibrant office or the comfort of your own home Opportunity to work abroad for up to two months per year Relocation opportunities within our offices in 55+ countries Corporate and social events We invest in your growth: Leadership development, career advising, soft skills and well-being programs Certifications, including GCP, Azure and AWS Unlimited access to EPAM's internal learning database Free English classes with certified teachers Discounts in local language schools, including online courses for the Kazakh language We cover it all: Participation in the Employee Stock Purchase Plan Monetary bonuses for engaging in the referral program Medical & family care package Six trust days per year (sick leave without a medical certificate) Coverage of psychology sessions of your choice Benefits package (sports activities, a variety of stores and services) Housing support program: preferential mortgage access via Otbasy Bank partnership Kyrgyzstan (Prod) We connect like-minded people: Delivering innovative solutions to industry leaders, making a global impact Enjoyable working environment, whether it is the vibrant office or the comfort of your own home Opportunity to work abroad for up to two months per year Relocation opportunities within our offices in 55+ countries Corporate and social events We invest in your growth: Leadership development, career advising, soft skills and well-being programs Certifications, including GCP, Azure and AWS Unlimited access to EPAM's internal learning database Free English classes with certified teachers We cover it all: Monetary bonuses for engaging in the referral program Medical & family care package Six trust days per year (sick leave without a medical certificate) Coverage of psychology sessions of your choice Discounts for fitness clubs and sports programs Benefits package (sports activities, a variety of stores and services) Uzbekistan (Prod) We connect like-minded people: Delivering innovative solutions to industry leaders, making a global impact Enjoyable working environment, whether it is the vibrant office or the comfort of your own home Opportunity to work abroad for up to two months per year Relocation opportunities within our offices in 55+ countries Corporate and social events We invest in your growth: Leadership development, career advising, soft skills and well-being programs Certifications, including GCP, Azure and AWS Unlimited access to EPAM's internal learning database Free English classes with certified teachers Discounts in local language schools, including offline courses for the Uzbek language We cover it all: Monetary bonuses for engaging in the referral program Medical & family care package Four trust days per year (sick leave without a medical certificate) Discounts for fitness clubs, dance schools and sports programs Benefits package (sports activities, a variety of stores and services)