Senior AI Engineer with Databricks

About the Position

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.

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 with a strong track record of delivery

Requirements

• 5+ 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
• Proven track record of running production-grade ML systems
• English proficiency at B2 level or higher

Nice to Have

• Experience with MLflow and MLOps practices
• Familiarity with AWS
• Background in Customer Analytics, Recommendation Systems, or Retail

We Offer

• 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), and benefits package (sports activities, a variety of stores and services)

About the Company

We connect like-minded people, delivering innovative solutions to industry leaders and making a global impact.