AI Architect - Databricks

Unison Group; Hyderabad; On-site; INR 3,500,000 - 7,500,000; Full time

Job summary

Unison Group in Hyderabad is seeking an experienced AI Architect to lead the design and implementation of scalable AI/ML platforms and intelligent data solutions. You will define enterprise AI architecture and drive data/AI transformation initiatives across business, engineering, and analytics teams.

The ideal candidate will have 8+ years in Data/AI architecture, deep Databricks expertise, and strong knowledge of end-to-end AI lifecycle, Lakehouse, and cloud services.

Qualifications

• Extensive hands-on experience designing enterprise AI architectures.
• Proven track record delivering scalable AI/ML platforms in cloud environments.
• Strong proficiency with Databricks and modern data platforms.

Responsibilities

• Design and implement enterprise AI/ML architecture on Databricks Lakehouse.
• Define scalable data ingest, feature engineering, training, deployment, monitoring, and governance.
• Architect and optimize AI/ML pipelines using Python, PySpark, SQL and cloud services.
• Lead MLOps/LLMOps frameworks for model lifecycle, CI/CD, registry, and deployment.
• Engage stakeholders to identify AI use cases including predictive analytics and NLP.
• Guide architecture for LLM/Generative AI, including RAG, vector stores and orchestration.
• Establish best practices for data quality, security, observability, and governance.
• Collaborate with data engineers, data scientists, product owners, and cloud teams.
• Provide technical leadership in AI tools, frameworks, and cloud services.

Skills

Databricks Lakehouse

Delta Lake

Unity Catalog

MLflow

Databricks Workflows

Model Serving

Feature Store

Python

PySpark

SQL

MLOps/LLMOps

Cloud platforms (Azure/AWS/GCP)

Generative AI/LLM ecosystems

RAG architectures

Vector stores

Data Engineering

ETL/ELT

Tools

Databricks Lakehouse

Delta Lake

MLflow

Job description

Job Description

We are looking for an experienced AI Architect with strong expertise in Databricks, Data Engineering, Machine Learning, and Cloud-based AI solutions to lead the design and implementation of scalable AI/ML platforms and intelligent data solutions. The ideal candidate will define enterprise AI architecture, drive modern data and AI transformation initiatives, and work closely with business, engineering, and analytics teams to deliver high-impact solutions.

This role requires a strong understanding of end-to-end AI/ML lifecycle, Lakehouse architecture, Databricks ecosystem, and the ability to translate business use cases into production-ready AI solutions.

Key Responsibilities

• Design and implement enterprise AI/ML architecture using Databricks Lakehouse Platform
• Define scalable solutions for data ingestion, feature engineering, model training, deployment, monitoring, and governance
• Architect and optimize AI/ML pipelines using Databricks, Spark, Python, SQL, and cloud-native services
• Lead the setup of MLOps / LLMOps frameworks for model lifecycle management, CI/CD, model registry, and automated deployment
• Work with business stakeholders to identify and prioritize AI/ML use cases, including predictive analytics, NLP, recommendation engines, and generative AI
• Build and guide architecture for LLM / Generative AI solutions, including RAG, vector databases, prompt orchestration, and model integration where applicable
• Establish best practices for data quality, security, compliance, observability, scalability, and responsible AI
• Collaborate with Data Engineers, Data Scientists, Product Owners, and Cloud teams to ensure solution alignment with enterprise architecture standards
• Provide technical leadership in selecting AI/ML tools, frameworks, and cloud services aligned to business and platform strategy
• Support architecture reviews, technical design workshops, PoCs, and enterprise AI roadmap planning

Required Skills & Experience

• 8- 15 years of experience in Data / AI / Analytics architecture, with strong exposure to enterprise-scale implementations
• Hands-on experience with Databricks including:
• Databricks Lakehouse
• Delta Lake
• Unity Catalog
• MLflow
• Databricks Workflows
• Model Serving / Feature Store (preferred)
• Strong programming experience in Python, PySpark, SQL, and AI/ML solution development
• Solid experience in designing and deploying Machine Learning pipelines in production
• Good understanding of MLOps / LLMOps, including model versioning, CI/CD, deployment, monitoring, and governance
• Experience with cloud platforms such as Azure, AWS, or GCP, preferably with Databricks integration
• Familiarity with Generative AI / LLM ecosystems, such as OpenAI, Hugging Face, LangChain, vector stores, embeddings, and RAG architectures
• Strong understanding of data engineering, ETL/ELT, distributed computing, and data platform modernization
• Experience in solution architecture, technical governance, and stakeholder engagement