AI/Machine Learning Engineer with Databricks

About the role

We are seeking a hands‑on AI/Machine Learning Engineer to design, build and operate optimized feature‑engineering, model‑training and inference pipelines. The role focuses on both batch and real‑time workloads, handling large‑scale structured and unstructured data.

Key responsibilities

• Design and optimise feature‑engineering pipelines for offline and online processing.
• Develop, deploy and maintain scalable model‑training and low‑latency inference pipelines.
• Implement streaming solutions using Spark Streaming for real‑time feature generation.
• Ensure reliability, performance and scalability of production ML systems.

Required profile

• 3+ years of experience building and operating offline and online feature‑engineering pipelines.
• Strong background in model training and real‑time inferencing.
• Proven ability to work with large‑scale unstructured data.
• English proficiency at B2 level or higher.

Required skills

• Databricks
• Apache Spark / PySpark
• Python
• SQL
• Spark Streaming

What we offer

• Fully remote work from any location in Armenia.
• Opportunities to work abroad up to two months per year.
• Leadership development, certifications and unlimited learning resources.
• Comprehensive medical and family care package, stock purchase plan and referral bonuses.