Lead AI Engineer (mlx, Agentic Ai, Gen AI Platform Services)

Capital One Capital One · Banking · New York, NY +4

Lead AI Engineer role focused on building and deploying AI-powered products and foundational AI systems, including LLM inference, similarity search, guardrails, and model evaluation, with an emphasis on optimizing performance and scalability for enterprise use.

What you'd actually do

  1. Design, develop, test, deploy, and support AI software components including foundation model training, large language model inference, similarity search, guardrails, model evaluation, experimentation, governance, and observability, etc.
  2. Leverage a broad stack of Open Source and SaaS AI technologies such as AWS Ultraclusters, Huggingface, VectorDBs, Nemo Guardrails, PyTorch, and more.
  3. Invent and introduce state-of-the-art LLM optimization techniques to improve the performance — scalability, cost, latency, throughput — of large scale production AI systems.
  4. Contribute to the technical vision and the long term roadmap of foundational AI systems at Capital One.

Skills

Required

  • Python
  • Go
  • Scala
  • Java
  • Computer Science
  • AI
  • Electrical Engineering
  • Computer Engineering

Nice to have

  • AWS
  • Google Cloud
  • Azure
  • Huggingface
  • VectorDBs
  • Nemo Guardrails
  • PyTorch
  • LLM Inference
  • Similarity Search
  • Guardrails
  • Memory
  • C++
  • C#
  • Golang
  • training optimization
  • inference optimization

What the JD emphasized

  • responsible and reliable AI systems
  • responsible and scalable ways
  • state-of-the-art LLM optimization techniques
  • scalable and responsible AI solutions

Other signals

  • foundation model training
  • large language model inference
  • similarity search
  • guardrails
  • model evaluation
  • experimentation
  • governance
  • observability