Senior AI Solution Engineer with Databricks, ML

We are seeking a Senior AI Solution Engineer with expertise in Databricks and Machine Learning to design, deploy, and operate production-grade ML systems. The ideal candidate is a hands-on Data Engineer / Machine Learning Engineer with a proven track record of delivering real-time and batch ML solutions, strong capability to build streaming pipelines, and experience ensuring the reliability and scalability of ML systems in production environments.

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 5+ 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 Familiarity with MLflow and MLOps practices Experience with AWS Background in Customer Analytics, Recommendation Systems, or Retail 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)