AI/ML Engineer (Databricks)
BengaluruFull-time3–6 yrsPosted 1 week ago
Mid-level data science role at Dysrupit in Bengaluru. Needs 3+ years.
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
- Machine learning
- Databricks
- LLMs
- Python
- Pandas
- Scikit-learn
Good to have
- SQL
- AWS
- Statistics
- Azure
- GCP
- MLOps