Staff AI Engineer (Remote Eligible)
$245k - $279k • Remote • San Francisco, CA • Full-time
Posted 1mo ago
About the job
Capital One is at the forefront of leveraging AI to transform banking, creating responsible and reliable AI systems that enhance customer experiences. The Intelligent Foundations and Experiences (IFX) team is central to this vision, building and deploying proprietary AI solutions that deliver value to millions. This role involves partnering with cross-functional teams to deliver AI-powered products, design and develop AI software components, and invent state-of-the-art foundation model optimization techniques. You will contribute to the technical vision and long-term roadmap of foundational AI systems, setting the technical direction for enterprise-wide AI architecture and championing responsible AI principles.
Responsibilities
- Partner with cross-functional teams to deliver AI-powered products.
- Design, develop, test, deploy, and support AI software components including foundation model training, LLM inference, agents, multi-agent workflows, similarity search, guardrails, model evaluation, experimentation, governance, and observability.
- Leverage Open Source and SaaS AI technologies such as AWS Ultraclusters, Huggingface, VectorDBs, and PyTorch.
- Invent and introduce state-of-the-art foundation model optimization techniques to improve performance (scalability, cost, latency, throughput) of large-scale production AI systems.
- Contribute to the technical vision and long-term roadmap of foundational AI systems.
- Set the technical direction for enterprise-wide AI architecture, unifying tooling, observability, and deployment standards.
- Own the design and integration of model routing, caching, and orchestration systems for hybrid and multi-model workloads.
- Champion responsible AI principles across all system design, embedding transparency, reproducibility, and fairness-by-design.
- Drive internal education, mentorship, and dissemination of best practices through architecture councils and AI guilds.
Requirements
- Bachelor's Degree in Computer Science, AI, Electrical Engineering, Computer Engineering, or related fields plus at least 8 years of experience developing AI and ML algorithms or technologies, or a Master's degree plus at least 6 years of experience.
- At least 8 years of experience programming with Python, Go, Scala, CUDA, or Java.
- Experience designing AI systems with tradeoff decisions around cost, latency, throughput and accuracy.
- 8+ years of experience deploying scalable and responsible AI solutions on cloud platforms (e.g. AWS, Google Cloud, Azure, or equivalent private cloud).
- Experience architecting, designing, developing, integrating, delivering, and supporting complex AI systems.
- Demonstrated ability to lead and mentor multiple engineering teams and influence cross-functional stakeholders up to the VP level.
- Experience developing AI and ML algorithms or technologies (e.g. LLM Inference, Similarity Search and VectorDBs, Guardrails, Memory) using Python, C++, C#, Java, CUDA, or Golang.
- Experience developing and applying state-of-the-art techniques for optimizing training and inference software to improve hardware utilization, latency, throughput, and cost.
- Experience in building agentic AI systems and agentic workflows.
- 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.
- Proven track record of defining and operationalizing enterprise AI architecture standards, including data pipeline governance, observability.