AI/Machine Learning Engineer with Databricks

We are looking for a hands-on AI/Machine Learning Engineer with proven experience deploying and running optimized feature engineering (offline and online), model training, and inference pipelines. Experience with unstructured data processing and Spark Streaming is a must. Real-time feature generation and inference are required for this project, along with batch pipelines for model training.
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 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
Requirements
3+ years of experience building, deploying, and operating offline and online feature engineering pipelines
Strong experience with model training and real-time inferencing pipelines
Hands-on expertise in Databricks, Spark/PySpark, Python, and SQL
Experience processing large-scale unstructured data
Strong knowledge of Spark Streaming
Experience with real-time feature generation and low-latency model inference
Experience building and supporting batch pipelines for model training and retraining
Proven track record of running production-grade ML systems
English proficiency at B2 level or higher
Nice to have
Experience with MLflow and MLOps
Experience with AWS
Background in Customer Analytics, Recommendation Systems, or Retail
We offer
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)
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.

Work arrangement
Yes