Hybrid Applied Researcher I – AI Foundations, LLM Core, Agentic AI

at Capital One

• Job title: Applied Researcher I – AI Foundations, LLM Core, Agentic AI
• Job type: Full Time
• Experience level: Junior, Mid level
• Salary: $218,700 - $272,300 per year
• Degree requirement: Postgraduate Degree
• Tech skills: AWS, Open Source, PyTorch
• Location requirements: Hybrid, United States

About the role

• Applied Researcher I collaborating with cross-functional teams to leverage AI technologies at Capital One. Engaging in impactful research to enhance customer interactions with banking services.

Responsibilities

• Partner with a cross-functional team of data scientists, software engineers, machine learning engineers and product managers to deliver AI-powered products that change how customers interact with their money
• Leverage a broad stack of technologies — Pytorch, AWS Ultraclusters, Huggingface, Lightning, VectorDBs, and more — to reveal the insights hidden within huge volumes of numeric and textual data
• Build AI foundation models through all phases of development, from design through training, evaluation, validation, and implementation
• Engage in high impact applied research to take the latest AI developments and push them into the next generation of customer experiences
• Flex your interpersonal skills to translate the complexity of your work into tangible business goals

Requirements

• Currently has, or is in the process of obtaining, a PhD in Electrical Engineering, Computer Engineering, Computer Science, AI, Mathematics, or related fields, with an exception that required degree will be obtained on or before the scheduled start date or M.S. in Electrical Engineering, Computer Engineering, Computer Science, AI, Mathematics, or related fields plus 2 years of experience in Applied Research
• PhD in Computer Science, Machine Learning, Computer Engineering, Applied Mathematics, Electrical Engineering or related fields
• LLM PhD focus on NLP or Masters with 5 years of industrial NLP research experience
• Multiple publications on topics related to the pre-training of large language models (e.g. technical reports of pre-trained LLMs, SSL techniques, model pre-training optimization)
• Member of team that has trained a large language model from scratch (10B + parameters, 500B+ tokens)
• Publications in deep learning theory
• Publications at ACL, NAACL and EMNLP, Neurips, ICML or ICLR
• Behavioral Models PhD focus on topics in geometric deep learning (Graph Neural Networks, Sequential Models, Multivariate Time Series)
• Multiple papers on topics relevant to training models on graph and sequential data structures at KDD, ICML, NeurIPs, ICLR
• Worked on scaling graph models to greater than 50m nodes
• Experience with large scale deep learning based recommender systems
• Experience with production real-time and streaming environments
• Contributions to common open source frameworks (pytorch-geometric, DGL)
• Proposed new methods for inference or representation learning on graphs or sequences
• Worked datasets with 100m+ users
• Optimization (Training & Inference) PhD focused on topics related to optimizing training of very large deep learning models
• Multiple years of experience and/or publications on one of the following topics: Model Sparsification, Quantization, Training Parallelism/Partitioning Design, Gradient Checkpointing, Model Compression
• Experience optimizing training for a 10B+ model
• Deep knowledge of deep learning algorithmic and/or optimizer design
• Experience with compiler design
• Finetuning PhD focused on topics related to guiding LLMs with further tasks (Supervised Finetuning, Instruction-Tuning, Dialogue-Finetuning, Parameter Tuning)
• Demonstrated knowledge of principles of transfer learning, model adaptation and model guidance
• Experience deploying a fine-tuned large language model

Benefits

• Comprehensive, competitive, and inclusive set of health, financial and other benefits that support your total well-being