Senior AI Engineer with Databricks

We are seeking a Senior AI Engineer with Databricks to design, deploy, and operate production-grade machine learning systems, with a strong focus on feature engineering, model training, and real-time inference pipelines. Experience the freedom of remote work from anywhere in Georgia, whether from the comfort of your home, our modern offices in Tbilisi and Batumi or a coworking space in Kutaisi. Responsibilities Design and optimize feature engineering pipelines for both batch and real-time workloadsBuild, deploy, and maintain scalable model training and inference pipelinesProcess structured and unstructured data at scaleDevelop streaming solutions using Spark StreamingEnable real-time feature generation and model servingEnsure reliability, scalability, and performance of ML solutions in productionSupport batch pipelines for model training and retrainingOperate and maintain production ML systems with a strong track record of delivery Requirements 5+ years of experience building, deploying, and operating offline and online feature engineering pipelinesStrong experience with model training and real-time inferencing pipelinesHands-on expertise in Databricks, Spark/PySpark, Python, and SQLExperience processing large-scale unstructured dataStrong knowledge of Spark StreamingExperience with real-time feature generation and low-latency model inferenceProven track record of running production-grade ML systemsEnglish proficiency at B2 level or higher Nice to have Experience with MLflow and MLOps practicesFamiliarity with AWSBackground in Customer Analytics, Recommendation Systems, or Retail We offer We connect like-minded peopleDelivering innovative solutions to industry leaders, making a global impactEnjoyable working environment, whether it is the vibrant office or the comfort of your own homeOpportunity to work abroad for up to two months per yearRelocation opportunities within our offices in 55+ countriesCorporate and social eventsWe invest in your growthLeadership development, career advising, soft skills and well-being programsCertifications, including GCP, Azure and AWSUnlimited access to EPAM's internal learning databaseFree English classes with certified teachersWe cover it allParticipation in the Employee Stock Purchase PlanMonetary bonuses for engaging in the referral programComprehensive medical & family care packageFive trust days per year (sick leave without a medical certificate)Benefits package (sports activities, a variety of stores and services) EPAM is global leader in AI transformation engineering and integrated consulting, serving Forbes Global 2000 companies and ambitious startups. With over thirty years of expertise in custom software, product and platform engineering, we empower our clients to become AI-Native enterprises, driving measurable value from innovation and digital investments.