Capital One

Lead AI Engineer

Capital One
Apply
10 days ago
Bengaluru, IndiaStaff+
H1B Sponsor

Responsibilities

  • Lead and deliver ML and NLP solutions in production environments.
  • Design and implement retrieval-augmented generation pipelines covering chunking, embedding, retrieval, and re-ranking.
  • Build, orchestrate, and evaluate AI agents with tool use and multi-step reasoning.
  • Train, fine-tune, and serve embedding and transformer-based models for search, classification, extraction, summarization, and generation.
  • Own productionization of ML and AI systems, including serving, monitoring, reliability, and drift management.
  • Optimize token usage and context windows for cost and performance.
  • Drive technical direction, AI engineering vision, and MLOps strategy, including standardized deployment, monitoring, and automated retraining frameworks.

Requirements

  • Bachelor’s degree in Computer Science or Engineering.
  • At least 8 years of experience with traditional machine learning algorithms, advanced NLP, model selection, experimentation, and evaluation.
  • At least 5 years of experience engineering and deploying production ML and AI systems, including high-throughput serving, ML continuous integration and delivery, latency optimization, and drift monitoring.
  • At least 3 years of experience with PyTorch, Hugging Face Transformers, and LangGraph for deep learning 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 developing LLM applications involving prompt engineering, model chaining, and external tool integration.
  • Preferred master’s degree in Computer Science or Engineering.
  • Preferred experience with statistics, probability, experimental design, hypothesis testing, A/B testing, causal inference, agentic search, multi-step query planning, retrieval optimization, LoRA, QLoRA, PEFT, RAGAS, DeepEval, BERT, GPT, Llama, Mistral, CrewAI, Claude Code, and OpenAI Codex.

Benefits

  • The position is based in Bangalore, Karnataka, India.
  • Capital One does not sponsor a new applicant for employment authorization for this position.
  • Capital One is an equal opportunity employer and provides reasonable workplace accommodations.

Tech Stack

Hugging Face TransformersPyTorch
Capital One

About Capital One

10,000+ employees

At Capital One, we're making things better for our customers and associates through innovation and collaboration. We were founded on the belief that everyone deserves financial freedom—and are dedicated to a world where all have equal opportunity to prosper. Banking is in our DNA, but we are so much more than a bank. We always think about what’s next—and how we can bring our customers the tools needed to improve their financial lives. Your ideas, experiences and skills will help make banking better. You’ll be part of a supportive culture while earning amazing benefits. That’s life at Capital One. Capital One is an equal opportunity employer (EOE, including disability/vet) committed to non-discrimination in compliance with applicable federal, state, and local laws. View our Social Media Community Guidelines https://www.capitalone.com/digital/social-media/

Contact me