Lead AI Engineer (mlx Emerging AI Patterns)

Capital One Capital One · Banking · San Jose, CA +1

Lead AI Engineer focused on designing, developing, testing, deploying, and supporting AI software components including foundation model training, LLM inference, similarity search, guardrails, model evaluation, experimentation, governance, and observability. The role involves leveraging AI technologies like AWS Ultraclusters, Huggingface, VectorDBs, and Nemo Guardrails, and inventing LLM optimization techniques for production AI systems. The candidate should have a strong engineering and mathematics foundation, expertise in hardware, software, and AI, and experience deploying scalable AI solutions on cloud platforms.

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
  • C++
  • C#
  • Golang
  • LLM Inference
  • Similarity Search
  • VectorDBs
  • Guardrails
  • Memory
  • training optimization
  • inference optimization

What the JD emphasized

  • foundation model training
  • large language model inference
  • similarity search
  • guardrails
  • model evaluation
  • governance
  • observability
  • LLM optimization techniques
  • scalable and responsible AI solutions

Other signals

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