
location_onNYU Paulson Center, 181, Mercer Street, University Village, Manhattan, New York County, New York, 10012, United States
At Capital One, we are creating trustworthy and reliable AI systems, changing banking for good. For years, Capital One has been leading the industry in using machine learning to create real-time, intelligent, automated customer experiences. 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.
The AI Foundations team is at the center of bringing our vision for AI at Capital One to life. Our work touches every aspect of the research life cycle, from partnering with Academia to building production systems. We work with product, technology, and business leaders to apply the state of the art in AI to our business, building world-class applied science and engineering teams with breakthrough product experiences and scalable, high-performance AI infrastructure.
Work model: On-site
NYU Paulson Center, 181, Mercer Street, University Village, Manhattan, New York County, New York, 10012, United States
New York, New York
Skills: Machine Learning, Pytorch, Aws, Huggingface, Lightning, Vectordbs, LLM, NLP, Deep Learning, Graph Neural Networks.
Education: PhD in Electrical Engineering, Computer Engineering, Computer Science, AI, Mathematics, or related fields required; Master's in Electrical Engineering, Computer Engineering, Computer Science, AI, Mathematics, or related fields with 4 years experience.
PhD in Computer Science, Machine Learning, Computer Engineering, Applied Mathematics, or Electrical Engineering. LLM focus on NLP (PhD) or Masters with 5 years of industrial NLP research experience. Multiple publications on pre-training of large language models (e.g., technical reports, SSL techniques, optimization). Membership in a team that trained a large language model from scratch (10B+ parameters, 500B+ tokens). Publications in deep learning theory. Publications at ACL, NAACL, EMNLP, Neurips, ICML, or ICLR. PhD focus on geometric deep learning (Graph Neural Networks, Sequential Models, Multivariate Time Series). Multiple papers on training models on graph and sequential data structures at KDD, ICML, NeurIPs, or ICLR. Experience 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 open-source frameworks (pytorch-geometric, DGL). Proposed new methods for inference or representation learning on graphs or sequences. Experience with datasets of 100m+ users. PhD focused on optimizing training of very large deep learning models. Multiple years of experience and/or publications on model sparsification, quantization, training parallelism/partitioning design, gradient checkpointing, or model compression. Experience optimizing training for a 10B+ model. Deep knowledge of deep learning algorithmic and/or optimizer design. Experience with compiler design. PhD focused on guiding LLMs with further tasks (Supervised Finetuning, Instruction-Tuning, Dialogue-Finetuning, Parameter Tuning). Demonstrated knowledge of transfer learning, model adaptation, and model guidance. Experience deploying a fine-tuned large language model.
Capital One • New York, New York