Lead AI Engineer (MLX Emerging AI Patterns)
Capital One4 days ago
San Jose, CA, USA or New York, NY, USAStaff+
Base Salary
$179k - $246k/yr
Responsibilities
- Partner with engineers, research scientists, technical program managers, and product managers to deliver AI-powered products.
- Design, develop, test, deploy, and support AI components covering foundation-model training, LLM inference, similarity search, guardrails, evaluation, experimentation, governance, and observability.
- Use open-source and SaaS AI technologies including AWS Ultraclusters, Hugging Face, VectorDBs, NeMo Guardrails, and PyTorch.
- Develop state-of-the-art LLM training and inference optimization techniques to improve scalability, cost, latency, throughput, and hardware utilization.
- Contribute to the technical vision and long-term roadmap for foundational AI systems.
Requirements
- Bachelor’s degree in Computer Science, AI, Electrical Engineering, Computer Engineering, or a related field plus at least four years of experience developing AI and ML algorithms or technologies, or a master’s degree in one of these fields plus at least two years of such experience.
- At least four years of programming experience with Python, Go, Scala, or Java.
- Preferred experience deploying scalable and responsible AI solutions on AWS, Google Cloud, Azure, or equivalent private cloud platforms.
- Preferred experience designing, developing, delivering, and supporting AI services.
- Preferred experience developing AI and ML technologies such as LLM inference, similarity search, VectorDBs, guardrails, and memory using Python, C++, C#, Java, or Golang.
- Preferred experience optimizing training and inference software for hardware utilization, latency, throughput, and cost.
- Passion for AI research and the ability to apply novel techniques in production.
Benefits
- Capital One offers health, financial, and other benefits supporting total well-being, with eligibility varying by employment status and management level.
- The role is eligible for performance-based incentive compensation, including cash bonuses and/or long-term incentives.
- Capital One will consider sponsoring a new qualified applicant for employment authorization for this position.
- Applications are expected to be accepted for a minimum of five business days.
About Capital One
Capital One is a U.S. consumer and commercial bank that offers credit cards, checking and savings accounts, auto financing, and lending to small and large businesses. It earns revenue from interest income and interchange/fees across its card and banking products, and develops cloud-based digital services. Founded in 1994, Capital One is publicly traded on the NYSE (COF) and is headquartered in McLean, Virginia.