Senior Lead AI Engineer (AI Foundations: LLM Customization, Finetuning, Reinforcement Learning)

Capital One; New York (NY); On-site; USD 251,000 - 286,000; Full time

Job summary

Capital One is seeking a Senior Lead AI Engineer to drive AI foundations, including model training and inference, on a high-performance platform. You will collaborate with engineers, researchers, PMs, and partners to deploy scalable AI capabilities across products and services.

You will work on optimization techniques for latency, cost, and throughput, while contributing to the roadmap of foundational AI systems at Capital One. The role supports innovation and responsible AI practices.

Qualifications

• Bachelor's degree in CS/AI/EE/CE or related field with 6+ years in AI/ML.
• Alternatively Master's in CS/AI/EE/CE with 4+ years in AI/ML.
• 6+ years of programming experience in Python, Go, Scala, or Java.

Responsibilities

• Partner with a cross-functional team to deliver AI-powered products.
• Design, develop, test, deploy, and support AI software components including training and inference.
• Leverage Open Source and SaaS AI technologies such as AWS Ultraclusters, Huggingface, VectorDBs, Nemo Guardrails, PyTorch.
• Invent and introduce state-of-the-art LLM optimization techniques for production systems.
• Contribute to the technical vision and long-term roadmap of foundational AI systems.

Skills

Python

Go

Scala

Java

LLM Inference

VectorDBs

PyTorch

HuggingFace

Guardrails

Memory

Education

Bachelor's degree in Computer Science, AI, Electrical Engineering, Computer Engineering, or related fields

Master's degree in Computer Science, AI, Electrical Engineering, Computer Engineering, or related fields

Tools

AWS

Google Cloud

Azure

Nemo Guardrails

VectorDBs

Job description

Senior Lead AI Engineer (AI Foundations: LLM Customization, Finetuning, Reinforcement Learning)

Overview: At Capital One, we are creating responsible and reliable AI systems, changing banking for good. For years, Capital One has been an industry leader in using machine learning to create real-time, personalized customer experiences. Our investments in technology infrastructure and world-class talent — along with our deep experience in machine learning — position us to be at the forefront of enterprises leveraging AI. From informing customers about unusual charges to answering their questions in real time, our applications of AI & ML are bringing humanity and simplicity to banking. We are committed to continuing to build world-class applied science and engineering teams to deliver our industry leading capabilities with breakthrough product experiences and scalable, high-performance AI infrastructure. At Capital One, you will help bring the transformative power of emerging AI capabilities to reimagine how we serve our customers and businesses who have come to love the products and services we build.

Team Description

The Intelligent Foundations and Experiences (IFX) team is at the center of bringing our vision for AI at Capital One to life. We work hand-in-hand with our partners across the company to advance the state of the art in science and AI engineering, and we build and deploy proprietary solutions that are central to our business and deliver value to millions of customers. Our AI models and platforms empower teams across Capital One to enhance their products with the transformative power of AI, in responsible and scalable ways for the highest leverage impact.

Responsibilities

• Partner with a cross-functional team of engineers, research scientists, technical program managers, and product managers to deliver AI-powered products that change how our associates work and how our customers interact with Capital One.
• 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.
• Leverage a broad stack of Open Source and SaaS AI technologies such as AWS Ultraclusters, Huggingface, VectorDBs, Nemo Guardrails, PyTorch, and more.
• Invent and introduce state-of-the-art LLM optimization techniques to improve the performance — scalability, cost, latency, throughput — of large scale production AI systems.
• Contribute to the technical vision and the long term roadmap of foundational AI systems at Capital One.

Ideal Candidate

You love to build systems, take pride in the quality of your work, and also share our passion to do the right thing. You want to work on problems that will help change banking for good. Passion for staying abreast of the latest research, and an ability to intuitively understand scientific publications and judiciously apply novel techniques in production. You adapt quickly and thrive on bringing clarity to big, undefined problems. You love asking questions and digging deep to uncover the root of problems and can articulate your findings concisely with clarity. You have the courage to share new ideas even when they are unproven. You are deeply Technical. You possess a strong foundation in engineering and mathematics, and your expertise in hardware, software, and AI enable you to see and exploit optimization opportunities that others miss. You are a resilient trail blazer who can forge new paths to achieve business goals when the route is unknown.

Basic Qualifications

• Bachelor's degree in Computer Science, AI, Electrical Engineering, Computer Engineering, or related fields plus at least 6 years of experience developing AI and ML algorithms or technologies, or a Master's degree in Computer Science, AI, Electrical Engineering, Computer Engineering, or related fields plus at least 4 years of experience developing AI and ML algorithms or technologies
• At least 6 years of experience programming with Python, Go, Scala, or Java

Preferred Qualifications

• 7 years of experience deploying scalable and responsible AI solutions on cloud platforms (e.g. AWS, Google Cloud, Azure, or equivalent private cloud)
• Experience designing, developing, integrating, delivering, and supporting complex AI systems
• Demonstrated ability to lead and mentor an engineering team and influence cross-functional stakeholders
• Experience developing AI and ML algorithms or technologies (e.g. LLM Inference, Similarity Search and VectorDBs, Guardrails, Memory) using Python, C++, C#, Java, or Golang
• Experience developing and applying state-of-the-0art techniques for optimizing training and inference software to improve hardware utilization, latency, throughput, and cost
• Passion for staying abreast of the latest AI research and AI systems, and judiciously apply novel techniques in production
• Excellent communication and presentation skills, with the ability to articulate complex AI concepts to peers

Salary (Annual)

• Cambridge, MA: $229,900 - $262,400 for Sr. Lead AI Engineer
• McLean, VA: $229,900 - $262,400 for Sr. Lead AI Engineer
• New York, NY: $250,800 - $286,200 for Sr. Lead AI Engineer
• San Francisco, CA: $250,800 - $286,200 for Sr. Lead AI Engineer
• San Jose, CA: $250,800 - $286,200 for Sr. Lead AI Engineer

This role is also eligible to earn performance based incentive compensation, which may include cash bonus(es) and/or long term incentives (LTI). Incentives could be discretionary or non discretionary depending on the plan.

Benefits

Capital One offers a comprehensive, competitive, and inclusive set of health, financial and other benefits that support your total well-being.

Practical Notes

Capital One will consider sponsoring a new qualified applicant for employment authorization for this position. The minimum and maximum full-time annual salaries for this role are listed below, by location. Please note that this salary information is solely for candidates hired to perform work within one of these locations, and refers to the amount Capital One is willing to pay at the time of this posting. Salaries for part-time roles will be prorated based upon the agreed upon number of hours to be regularly worked. Candidates hired to work in other locations will be subject to the pay range associated with that location, and the actual annualized salary amount offered to any candidate at the time of hire will be reflected solely in the candidate’s offer letter. This role is expected to accept applications for a minimum of 5 business days. Capital One is an equal opportunity employer (EOE, including disability/vet) committed to non-discrimination in compliance with applicable federal, state, and local laws. Capital One promotes a drug-free workplace.

Capital One does not provide, endorse nor guarantee and is not liable for third-party products, services, educational tools or other information available through this site.

At Capital One, we’re building a leading information-based technology company. Still founder-led by Chairman and Chief Executive Officer Richard Fairbank, Capital One is on a mission to help our customers succeed by bringing ingenuity, simplicity, and humanity to banking. We measure our efforts by the success our customers enjoy and the advocacy they exhibit.