Databricks AI Architect

Persistent; Pune District; Remote; INR 4,200,000 - 6,400,000; Full time

Benefits offered by this job

Competitive salary and benefits

Growth opportunities and education

Cutting-edge technologies

Employee engagement initiatives

Health check-ups

Insurance coverage

Job summary

Persistent seeks an experienced Databricks AI Architect to design and implement enterprise-scale AI, ML, and Generative AI solutions. You will drive scalable AI platforms, architect RAG and LLM workflows, and mentor cross-functional teams across data engineering and AI engineering.

You will partner with stakeholders to translate requirements into technical AI solutions, oversee governance, and advance responsible AI practices while staying current with MLOps and cloud-native AI trends.

Qualifications

• Hands-on Databricks Lakehouse experience with scalable AI platforms.
• Strong Python and PySpark development skills.
• Experience building enterprise ML solutions and MLOps.
• Knowledge of Generative AI, LLMs, and RAG frameworks.
• Ability to translate business needs into technical AI designs.
• Proven collaboration with data engineering and business stakeholders.

Responsibilities

• Design and deploy end-to-end ML and Generative AI solutions using Databricks.
• Develop scalable feature engineering and data pipelines.
• Manage ML lifecycle, experiment tracking, and model registry with MLflow.
• Architect RAG-based knowledge retrieval across enterprise data sources.
• Build intelligent assistants and knowledge search tools.
• Lead architecture discussions and set AI best practices.

Skills

Databricks Lakehouse

Python

PySpark

Machine Learning

Generative AI

AI architecture

MLOps

LLMs

RAG

Stakeholder management

Mentoring

Tools

MLflow

Vector databases

Embeddings

OpenAI API

Azure OpenAI

Job description

AboutPosition:

We are seeking an experienced Databricks AI Architect to lead the design and implementation of enterprise-scale Artificial Intelligence, Machine Learning, and Generative AI solutions. The ideal candidate will possess deep expertise in Databricks, Python, PySpark, Machine Learning, and modern AI architectures. This role involves architecting scalable AI platforms, driving innovation using Generative AI and Agentic AI technologies, and delivering business-focused AI solutions that leverage large-scale data ecosystems.

• Role: Databricks AI Architect


• Location: All Persistent Locations


• Experience: Between 11 to 16 Years


• Job Type: Full Time Employment

WhatYou'llDo:

• Design, develop, and deploy end-to-end Machine Learning and Generative AI solutions using Databricks, Python, and PySpark.


• Perform Exploratory Data Analysis (EDA) to identify data patterns, trends, quality issues, and actionable insights.


• Build, train, evaluate, optimize, and deploy Machine Learning models for predictive and prescriptive analytics.


• Design scalable feature engineering, data preparation, and model training pipelines using Databricks and PySpark.


• Implement and manage ML lifecycle processes using MLflow, including experiment tracking, model registry, versioning, and deployment.


• Architect Retrieval-Augmented Generation (RAG) solutions using enterprise data sources, vector databases, embeddings, and Large Language Models (LLMs).


• Develop Generative AI applications such as intelligent assistants, chatbots, knowledge search platforms, and document intelligence solutions.


• Design enterprise AI architectures that ensure scalability, security, reliability, and operational excellence.


• Lead architecture discussions and define best practices for AI, ML, and data platform implementations.


• Evaluate and benchmark AI models for accuracy, performance, explainability, and business impact.


• Collaborate with business stakeholders to translate functional requirements into technical AI solutions.


• Drive AI governance, model management, responsible AI practices, and deployment standards.


• Support data engineering initiatives including ingestion, transformation, optimization, and pipeline development within the Databricks ecosystem.


• Contribute to Agentic AI initiatives involving autonomous workflows, intelligent agents, and AI-driven automation.


• Mentor engineering and data science teams on AI architecture, model development, and best practices.


• Stay current with advancements in Machine Learning, Generative AI, Agentic AI, LLMs, MLOps, and cloud-native AI technologies.

ExpertiseYou'llBring:

• 11+ years of experience in Data Science, Machine Learning, AI Engineering, or Data Platform Architecture.


• Strong hands-on expertise with Databricks and the Databricks Lakehouse Platform.


• Deep expertise in Python and PySpark development.


• Strong understanding of Machine Learning algorithms, model evaluation techniques, and statistical analysis.


• Experience building and deploying enterprise-grade Machine Learning solutions.


• Hands-on experience performing Exploratory Data Analysis (EDA) and feature engineering on large datasets.


• Strong understanding of data engineering concepts and distributed data processing.


• Experience using MLflow for experiment management, model lifecycle management, and deployment workflows.


• Knowledge of MLOps principles, model monitoring, and production AI operations.


• Experience designing and implementing Data Lakehouse architectures.


• Strong SQL and data manipulation capabilities.


• Hands-on experience designing and developing Generative AI solutions.


• Strong knowledge of Large Language Models (LLMs) and enterprise AI architectures.


• Experience implementing Retrieval-Augmented Generation (RAG) frameworks and knowledge retrieval systems.


• Understanding of vector databases, embeddings, semantic search, and retrieval optimization.


• Experience integrating AI services such as OpenAI, Azure OpenAI, Anthropic Claude, Gemini, or similar platforms.


• Understanding of Agentic AI concepts and autonomous AI workflows.


• Practical implementation experience is preferred; strong theoretical knowledge is also acceptable.


• Experience with prompt engineering, prompt optimization, and LLM evaluation frameworks.


• Experience architecting enterprise-scale AI and Machine Learning platforms.


• Ability to define technical roadmaps, architecture standards, and implementation strategies.


• Strong stakeholder management and solution design capabilities.


• Experience leading cross-functional teams across Data Engineering, AI Engineering, and Business functions.


• Excellent communication, presentation, and mentoring skills.

Benefits:

• Competitive salary and benefits package


• Culture focused on talent development with quarterly growth opportunities and company-sponsored higher education and certifications


• Opportunity to work with cutting-edge technologies


• Employee engagement initiatives such as project parties, flexible work hours, and Long Service awards


• Annual health check-ups


• Insurance coverage: group term life, personal accident, and Mediclaim hospitalization for self, spouse, two children, and parents

Values-Driven,People-Centric&InclusiveWorkEnvironment:

Persistent is dedicated to fostering diversity and inclusion in the workplace. We invite applications from all qualified individuals, including those with disabilities, and regardless of gender or gender preference. We welcome diverse candidates from all backgrounds.

• We support hybrid work and flexible hours to fit diverse lifestyles.


• Our office is accessibility-friendly, with ergonomic setups and assistive technologies to support employees with physical disabilities.


• If you are a person with disabilities and have specific requirements, please inform us during the application process or at any time during your employment.

"Persistent is an Equal Opportunity Employer and prohibits discrimination and harassment of any kind."