Lead Machine Learning Engineer

Lead Machine Learning Engineer role at Capital One in Risk Tech focuses on designing, deploying, and maintaining AI and ML systems while delivering AI powered products with an emphasis on responsible and explainable AI.

Responsibilities

• Collaborate with engineers, data scientists, product managers, and designers to deliver AI powered solutions that improve associates' workflows and customer value.
• Design, develop, test, deploy, and support AI software components using machine learning models, including model evaluation, experimentation, large language model inference, similarity search, guardrails, governance, observability, and agentic AI.
• Fine tune, develop, and evaluate machine learning and foundational models.
• Work within a cross functional Agile team to build and enhance software that leverages advanced AI and ML capabilities.
• Contribute thought leadership and technical vision to the long term AI roadmap at Capital One.
• Leverage a broad mix of Open Source and SaaS AI technologies.
• Inform ML infrastructure decisions based on an understanding of modeling techniques and associated challenges.
• Retrain, maintain, and monitor models in production environments.
• Build optimized data pipelines to feed ML models.
• Ensure code quality reduces vulnerabilities, maintain model governance from a risk perspective, and follow Responsible and Explainable AI practices.

Requirements

• Bachelor’s Degree
• At least 6 years of experience designing and building data intensive solutions using distributed computing (Internship experience does not apply)
• At least 4 years of experience programming with Python, Scala, or Java
• At least 2 years of experience building, scaling, and optimizing ML systems

Technologies

• Python
• Scala
• Java
• scikit-learn
• PyTorch
• Dask
• Spark
• TensorFlow
• AWS Bedrock
• Google Cloud
• Azure
• Retrieval Augmented Generation (RAG)

Benefits

• Health benefits
• Financial benefits
• Other benefits
• Performance-based incentive compensation (cash bonus(es) and/or long term incentives)

The ideal candidate

• You love building systems, take pride in quality, and share a commitment to doing the right thing while tackling problems that can transform banking.
• You are a strong communicator who can explain complex technical concepts to non technical partners across the business, including presentations to large audiences.
• You stay current with AI research and can interpret scientific publications to apply novel techniques in production.
• You adapt quickly, bring clarity to large, undefined problems, ask questions, and dig deep to uncover root causes, articulating findings concisely; you are willing to share new ideas even when unproven.
• You are deeply technical with a solid foundation in engineering and mathematics, and you can spot optimization opportunities across hardware, software, and AI.
• You are a resilient trailblazer who can forge new paths to meet business goals when the path is unclear.
• You maintain a strong appetite for staying current with AI research and systems, applying novel techniques in production in a prudent manner.