AI/Machine Learning Engineer with Databricks

About the Position

We are seeking a Senior AI/Machine Learning Engineer with expertise in Databricks for a remote position, focusing on feature engineering and model training.

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
• 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, Discounts for fitness clubs and sports programs, Benefits package (sports activities, a variety of stores and services)
• Housing support program: preferential mortgage access via Otbasy Bank partnership

About the Company

We are a global team of innovators, delivering cutting-edge solutions to industry leaders and making a significant global impact. Our culture values collaboration, creativity, and continuous learning, providing an enjoyable working environment whether you prefer the office or working from home. We offer opportunities for professional growth, including leadership development, career advising, and access to certifications. Our comprehensive benefits package includes medical and family care, participation in the Employee Stock Purchase Plan, and various discounts and perks.