New opportunity

Lead AI/ML Engineer

LeadSquared · Bangalore, India

About this role

The Lead AI/ML Engineer will design, build, and deploy production-grade AI systems, ranging from agentic LLM applications to classical machine learning models. You will own end-to-end solutions, lead a small team, and translate business requirements into reliable AI products. Responsibilities include architecting agentic workflows with planning, tool calling, and memory; optimizing RAG pipelines including chunking, embeddings, and re-ranking; and deploying models using MLOps best practices. You will also mentor team members and collaborate with product and business stakeholders. Required qualifications include 4-6 years of overall experience, with at least 1-2 years specifically in LLM or GenAI development. Candidates must possess expert-level Python skills and hands-on experience with LangChain, LangGraph, and vector databases. A degree in Computer Science, AI, or Data Science is required. Preferred qualifications include experience with LLM fine-tuning (LoRA/QLoRA), evaluation frameworks like LangSmith or Ragas, and MLOps tools such as MLflow or SageMaker.

For full details about this role, click “Apply on company site” to view the employer’s original posting.

Skills for this role

PythonNumPyPandasFastAPIAgentic AILangChainLangGraphRAGPineconeWeaviateQdrantMilvusChromaFAISSscikit-learnPyTorchTensorFlowGitDockerAWSAzureGCP

Your skill match

Checking your profile…

YOUR NEXT STEP

Get interview-ready for this role

A focused preparation guide, built around this job’s responsibilities and requirements.

✦ AI-generated guide
Preparation suggestions, not the employer’s actual interview questions. Always check the original posting for current requirements.
01

Understand the role

What matters most

  • Architecting Agentic Workflows

    The role requires designing systems that incorporate planning, tool calling, and memory using LangChain and LangGraph. You must demonstrate the ability to move beyond simple prompt engineering to build complex, multi-agent systems that can reliably execute multi-step tasks.

  • Optimizing RAG Pipelines

    You are expected to own the end-to-end RAG lifecycle, including chunking strategies, embedding selection, vector search optimization, and re-ranking. Success depends on your ability to evaluate and improve retrieval accuracy for production-grade applications.

  • Technical Leadership and Mentorship

    As a lead, you must balance hands-on development with guiding a small team. This involves reviewing code, providing architectural direction, and translating high-level business requirements into technical specifications for your team to implement.

  • Production-Grade MLOps

    The role demands moving models from experimentation to deployment. You must be proficient in exposing models via FastAPI and applying MLOps best practices, such as containerization with Docker and cloud-based deployment, to ensure system reliability.

Likely challenges to think through

  • Balancing Agentic Complexity with Reliability

    Agentic workflows using LangGraph can become non-deterministic and difficult to debug. Prepare by discussing your strategies for implementing guardrails, logging, and observability to ensure these systems remain stable in a production environment.

  • Bridging Traditional ML and GenAI

    The role requires expertise in both classical ML (scikit-learn, PyTorch) and modern LLM frameworks. You may face the challenge of integrating these disparate systems; prepare by highlighting instances where you have combined traditional predictive modeling with generative AI outputs.

  • Stakeholder Alignment on AI Capabilities

    Translating business problems into AI products often involves managing expectations regarding model performance and latency. Prepare by explaining how you communicate technical constraints and trade-offs to non-technical product and business stakeholders.

03

Practice your answers

Technical practice

Architecting Agentic Workflows

What to coverExplain how you design multi-agent systems using LangGraph to handle complex planning and tool-calling tasks. Discuss how you manage state, memory, and error recovery in these workflows to ensure reliability in production.

Optimizing RAG Pipelines

What to coverDescribe your approach to improving RAG performance beyond basic retrieval. Address specific techniques for chunking strategies, embedding selection, and the implementation of re-ranking mechanisms to increase precision.

Production-Grade Python and API Design

What to coverDiscuss your process for building scalable, production-ready AI services using FastAPI. Focus on handling asynchronous operations, managing concurrency, and ensuring robust error handling for model inference endpoints.

Vector Database Selection and Scaling

What to coverCompare different vector databases like Pinecone, Milvus, or Qdrant based on performance, scalability, and cost. Explain how you choose the right indexing strategy for a specific use case to balance search latency and accuracy.

LLM Evaluation Frameworks

What to coverHow do you systematically evaluate LLM outputs in a production environment? Suggest using frameworks like Ragas or LangSmith to measure metrics such as faithfulness, answer relevance, and context precision.

Fine-Tuning Strategies (LoRA/QLoRA)

What to coverExplain the trade-offs between full fine-tuning and parameter-efficient methods like LoRA or QLoRA. Discuss when it is appropriate to fine-tune a model versus relying on RAG or prompt engineering.

MLOps and Model Deployment

What to coverDescribe your workflow for deploying models using Docker and cloud-native tools like SageMaker or MLflow. Focus on versioning, monitoring for model drift, and automating the CI/CD pipeline for AI models.

Classical ML vs. GenAI Integration

What to coverDiscuss scenarios where traditional machine learning models (scikit-learn) are more effective than LLMs. Explain how you integrate these different paradigms into a single, cohesive product architecture.

Behavioural practice

Leading and Mentoring Technical Teams

What to coverDescribe a situation where you had to mentor a junior engineer or lead a technical project. Focus on how you balanced hands-on development with code reviews and design guidance to ensure team growth and project success.

Translating Business Requirements to AI Solutions

What to coverProvide an example of how you translated a vague business problem into a concrete AI product. Explain how you communicated technical trade-offs to non-technical stakeholders to manage expectations.

Handling Production Failures

What to coverDescribe a time when a production AI system failed or performed unexpectedly. Detail the steps you took to diagnose the issue, the immediate mitigation, and the long-term architectural changes you implemented to prevent recurrence.

Managing Technical Debt in AI Projects

What to coverExplain how you balance the need for rapid prototyping in AI with the requirement for maintainable, production-grade code. Give an example of how you prioritized refactoring or infrastructure improvements alongside new feature development.

02

Sharpen your knowledge

Topics to prepare

  • Agentic Workflow Architecture

    Focus on designing systems that incorporate planning, tool calling, and memory. Exercise: Draft a system architecture diagram for a multi-agent workflow that uses LangGraph to handle a complex user query requiring multiple tool interactions and state persistence. This is directly relevant to the requirement of building agentic AI systems.

  • RAG Pipeline Optimization

    Study the end-to-end RAG lifecycle including chunking strategies, embedding models, vector search, and re-ranking. Exercise: Compare the impact of different chunking sizes and re-ranking algorithms on retrieval precision for a sample dataset. This aligns with the responsibility to architect and optimize RAG pipelines.

  • MLOps and Deployment Best Practices

    Understand the lifecycle of production-grade AI, from model training to deployment and monitoring. Exercise: Outline a CI/CD pipeline for an LLM application using Docker and a cloud provider, incorporating model versioning and evaluation metrics. This is essential for the role's focus on shipping production-grade systems.

Skills to revise

  • LangChain and LangGraph Implementation

    Develop proficiency in building stateful applications. Exercise: Create a small LangGraph project that implements a cyclical workflow with conditional logic and memory. This is a core requirement for the advertised role.

  • Vector Database Integration

    Gain hands-on experience with vector search operations. Exercise: Implement a script to ingest documents into a vector database (e.g., Pinecone or Chroma), perform similarity searches, and integrate a re-ranking step. This directly addresses the requirement for practical experience with vector databases.

  • LLM Evaluation Frameworks

    Learn to measure the performance of GenAI applications. Exercise: Use Ragas or LangSmith to evaluate a RAG pipeline's faithfulness and answer relevance against a set of ground-truth questions. This is a preferred skill that demonstrates technical maturity in AI product development.

04

Your preparation checklist

  • Prepare an agentic workflow case study

    Document a project where you implemented an agentic system using LangChain or LangGraph. Focus on how you handled planning, tool calling, and memory management to solve a specific business problem.

  • Demonstrate RAG pipeline optimization

    Prepare a technical walkthrough of a RAG system you built. Be ready to explain your choices for chunking strategies, embedding models, vector database selection, and how you implemented re-ranking to improve retrieval accuracy.

  • Review MLOps and deployment practices

    Outline your experience with productionizing models. Prepare to discuss how you use Docker, Git, and cloud infrastructure (AWS/Azure/GCP) to maintain model performance, including your approach to monitoring and evaluation using tools like LangSmith or MLflow.

  • Refine your leadership and mentorship narrative

    Prepare specific examples of how you have led small teams, conducted code reviews, and translated complex business requirements into actionable technical tasks for other engineers.

  • Formulate strategic questions for the interviewers

    Prepare questions regarding the current maturity of their AI products, the specific business problems they are prioritizing for agentic workflows, and how the team balances rapid experimentation with production-grade stability.

AI-generated from the job information · Prepared Oct 30, 2026 (UTC)Useful preparation, not a guarantee of eligibility or an offer.

Availability has not yet been verified. Status can change after a check; confirm on the employer’s website. Report an issue · Our checking policy

KEEP EXPLORING

Similar AI jobs

Related skills and specializations in India. Matched to this role, not your profile.

Explore more jobs
Finding similar opportunities…