Senior AI Engineer with Databricks

About the role

We are looking for a Senior AI Engineer with deep expertise in Databricks to design, deploy, and operate production‑grade machine learning systems. The position is fully remote, allowing you to work from anywhere in Armenia.

Key responsibilities

• Design and optimise feature‑engineering pipelines for batch and real‑time workloads.
• Build, deploy and maintain scalable model training and inference pipelines.
• Process structured and unstructured data at scale using Spark and Spark Streaming.
• Develop streaming solutions and enable real‑time feature generation and low‑latency model serving.
• Ensure reliability, scalability and performance of ML solutions in production.

Required profile

• 5+ years of experience building, deploying and operating offline and online feature‑engineering pipelines.
• Strong track record of running production‑grade ML systems.
• English proficiency at B2 level or higher.

Required skills

• Databricks
• Spark / PySpark
• Python
• SQL
• Spark Streaming
• MLflow (nice to have)
• AWS (nice to have)

What we offer

• Flexible remote work from any location in Armenia.
• Opportunities to work abroad for up to two months per year.
• Leadership development, career advising and well‑being programs.
• Certifications in GCP, Azure and AWS.
• Comprehensive medical and family care package.
• Employee Stock Purchase Plan and referral bonuses.