AI/ML Engineer (Databricks)

About the role

We are seeking an AI/ML Engineer to deliver production-grade AI/ML solutions on Databricks for enterprise clients. You will build robust pipelines, productionize machine learning and Generative AI use cases, and help clients move from notebook experimentation to reliable, governed, at-scale production systems.

What you'll do

• Build and productionize feature pipelines and training/serving pipelines for ML and GenAI use cases, using MLflow for experiment tracking and model management
• Support the delivery of Generative AI and agentic systems on Databricks, including data preparation for RAG pipelines, vector search integration
• Help stand up new AI use cases end to end on Databricks, from data preparation through to a working prototype
• Build reusable Databricks templates and accelerators for common AI/ML use cases (RAG, agentic assistants, forecasting, personalisation)
• Optimize Databricks jobs, clusters, and workflows for cost, performance, and reliability at production scale

What you need

• 3–6 yrs of experience
• Machine learning
• Databricks
• LLMs
• Python
• Pandas
• Scikit-learn
• Any graduate

Good to have

• SQL
• AWS
• Statistics
• Azure
• GCP
• MLOps