Lead AI Engineer

Capital One Capital One · Banking · Bangalore, IN

Lead AI Engineer role focused on building and deploying production ML/NLP solutions, leveraging LLMs for workflows, designing RAG pipelines, building and evaluating AI agents with tool use, and optimizing serving infrastructure. The role emphasizes technical direction, MLOps strategy, and productionization of AI systems.

What you'd actually do

  1. Lead and deliver ML/NLP solutions in production environments
  2. Leverage LLMs to drive and enhance ML/DS workflows
  3. Design and implement Retrieval-Augmented Generation pipelines covering chunking, embedding, retrieval and re-ranking
  4. Build, orchestrate and evaluate AI agents with tool use and multi-step reasoning
  5. Define and execute evaluation frameworks for generative AI systems

Skills

Required

  • Machine learning algorithms
  • Advanced natural language processing
  • Model selection
  • Machine learning experimentation lifecycle
  • Offline or online evaluation
  • Engineering and deploying production machine learning and AI systems
  • High-throughput model serving
  • Continuous integration and delivery for machine learning
  • Latency optimization
  • Continuous drift monitoring
  • PyTorch
  • Hugging Face Transformers
  • LangGraph
  • Deep learning models
  • Agentic workflows
  • Developing, fine-tuning, and serving embedding models
  • Sentence-transformers
  • Representation learning
  • Large Language Model application development
  • Advanced prompt engineering
  • Model chaining architectures
  • External tool integration

Nice to have

  • Statistics
  • Probability theory
  • Experimental design
  • Hypothesis testing
  • A/B testing frameworks
  • Causal inference methodologies
  • Agentic search
  • Multi-step LLM driven query planning
  • Retrieval pipelines
  • Bi-encoders
  • Cross-encoders
  • Hybrid search techniques
  • Dense and sparse search
  • Fine-tuning pipelines
  • LoRA
  • QLoRA
  • PEFT
  • AI evaluation frameworks
  • RAGAS
  • DeepEval
  • Custom quantitative metrics
  • Transformer architectures
  • Encoder paradigms
  • Decoder paradigms
  • BERT
  • GPT
  • Llama
  • Mistral
  • Multi-agent frameworks
  • Stateful agentic memory architectures
  • AI-powered coding environments
  • CrewAI
  • Claude Code
  • OpenAI Codex

What the JD emphasized

  • At least 8 years of experience in traditional machine learning algorithms, advanced natural language processing , model selection and the machine learning experimentation lifecycle including baseline modeling, iterative improvement and offline or online evaluation
  • At least 5 years of experience engineering and deploying production machine learning and AI systems, including high-throughput model serving, continuous integration and delivery for machine learning, latency optimization and continuous drift monitoring
  • At least 3 years of experience leveraging PyTorch, Hugging Face Transformers, and LangGraph to develop deep learning models and agentic workflows
  • At least 3 years of experience developing, fine-tuning, and serving embedding models using sentence-transformers and modern representation learning stacks
  • At least 2 years of experience in Large Language Model application development, specializing in advanced prompt engineering, model chaining architectures, and external tool integration

Other signals

  • LLMs
  • AI Agents
  • RAG
  • Production ML