Technical Lead - AI ML

UST; Bengaluru; Hybrid; INR 4,200,000 - 6,800,000; Full time

Job summary

UST is seeking a Lead AI Engineer in Bengaluru to lead ML/AI engineering with a strong focus on production-grade LLM/agentic solutions. The role requires mentoring a team, delivering scalable Python-based AI systems, and leveraging Azure AI, Databricks ML, and modern MLOps practices.

The candidate should demonstrate expertise in LLM prompt engineering, vector retrieval, and end-to-end AI deployments that create measurable business impact for global clients.

Qualifications

• 5+ years in ML/AI engineering with production LLM/agentic delivery.
• Advanced Python and at least one agent framework (LangGraph, AutoGen, CrewAI, PydanticAI).
• Strong LLM and prompt-engineering skills (GPT, Claude, LLaMA); hands-on RAG workflows.
• Azure AI stack (Azure OpenAI, AI Search, AI Services) and Databricks ML (MLflow, Delta Lake).
• Vector databases (FAISS, Pinecone, Chroma) and embedding/retrieval design.
• Containerized deployment (Docker/Kubernetes, AKS/ARO) and REST APIs / WebSockets / event-driven services.
• CI/CD and version control (Jenkins / GitHub Actions, Git) with SDLC and agile practices.
• Portfolio of 3+ production AI deployments with measurable business impact.

Skills

LLM delivery

Python

Agent frameworks

LLM prompt engineering

Azure AI

Databricks ML

Vector databases

Embeddings design

Docker/Kubernetes

REST/WebSockets

CI/CD

Production AI deployments

Education

B.E./B.Tech or M.Tech in CS/Data Science

Tools

LangGraph

AutoGen

CrewAI

PydanticAI

Jenkins

GitHub Actions

Git

AKS/ARO

Job description

At UST, we help the worlds best organizations grow and succeed through transformation. Bringing together the right talent, tools, and ideas, we work with our client to co-create lasting change. Together, with over 30,000 employees in 30+ countries, we build for boundless impacttouching billions of lives in the process. Visit us at UST.com.

Lead AI Engineer

Experience: 5+ years in ML/AI engineering with Minimum 1 year of exp in mentoring team.

Education: B.E./B.Tech or M.Tech in CS, Data Science, or related field

Must-Have Skills

• 5+ years ML/AI engineering with production LLM/agentic delivery.
• Advanced Python and at least one agent framework (LangGraph, AutoGen, CrewAI, PydanticAI).
• Strong LLM and prompt-engineering skills (GPT, Claude, LLaMA); hands-on RAG workflows.
• Azure AI stack (Azure OpenAI, AI Search, AI Services) and Databricks ML (MLflow, Delta Lake).
• Vector databases (FAISS, Pinecone, Chroma) and embedding/retrieval design.
• Containerized deployment (Docker/Kubernetes, AKS/ARO) and REST APIs / WebSockets / event-driven services.
• CI/CD and version control (Jenkins / GitHub Actions, Git) with SDLC and agile practices.
• Portfolio of 3+ production AI deployments with measurable business impact.