Applied AI ML Lead [Multiple Positions Available]

JPMorgan Chase & Co.•Columbus, OH

6d•Onsite

About The Position

This role involves engaging in cutting-edge initiatives to enhance virtual assistant capabilities by leveraging state-of-the-art Natural Language Processing (NLP), deep learning, and generative AI techniques. The position drives the development of scalable, production-grade AI solutions through experimentation with large language models (LLMs) and small language models (SLMs), including domain-specific fine-tuning and optimization. The lead will also be involved in all aspects of machine learning, supporting tech and product teams, and providing expertise in ML applications. A key responsibility includes owning the machine learning solution for the transaction search application and leading the build of a question-and-answer solution for the Chase.com website using advanced NLP techniques. The role requires conducting thorough analysis of business needs, exploring state-of-the-art research, and designing/executing experiments with LLMs and SLMs. The lead will also manage model releases and drive project success to deliver production-ready solutions.

Requirements

Bachelor's degree in Computer Engineering, Computer Science, Information Technology, or a related field of study plus 7 years of experience in the job offered or a related occupation.
Alternatively, a Master's degree in Computer Engineering, Computer Science, Information Technology, or a related field of study plus 5 years of experience in the job offered or a related occupation.
Utilizing Python to implement data science solutions, build scalable machine learning (ML) pipelines, and automate workflows.
Applying Supervised and Unsupervised Learning to build predictive ML models, improve decision-making, and automate labeling.
Using feature engineering to identify and select relevant features to improve ML model performance.
Leveraging Hyperparameter Optimization to enhance ML model accuracy and generalization.
Utilizing Neural Networks including Convolution Neural Networks (CNN), Recurrent Neural Networks (RNN), Long Short-Term Memory's (LSTM), and Transformers to build ML solutions, automate tasks, and fine-tune domain-specific language models.
Using Open Source Embedding Models including transformers (all-mpnet-base-v2) and sentence transformers (msmarco-distilbert-base-tas-b) to capture semantic and contextual relationships in text data.
Applying Tokenization, Named Entity Recognition, Semantic Search, and Topic Modeling to structure and analyze text data.
Using Prompt Engineering, System Prompt Design, Retrieval-Augmented Generation (RAG), Instruction Fine-Tuning, Parameter-Efficient Fine-Tuning, Multi-adapter Architectures, Domain Adaptation model training for Banking and Financial NLP, Synthetic Data Generation for Fine-Tuning and to Enhance LLM performance.
Utilizing Dense Retrieval, Sparse Retrieval, Hybrid Search, Embedding-based Semantic Search to improve information retrieval accuracy and efficiency.
Using Precision, Recall, F1, Mean Reciprocal Rank (MRR), Normalized Discounted Cumulative Gain (NDCG), SQUAD Metrics, Exact Match, Perplexity, Multi-class and Multi-label Evaluation, Latency Profiling, Human-in-the-loop Evaluation to assess and validate ML model effectiveness and performance.
Utilizing Distributed Training using Data Parallel, Fully Sharded Data Parallel, Mixed Precision Training, Multi-GPU Scaling for LoRA (Low Rank Adapters), Fine-Tuning of SLMs (Small Language Models) to scale SLMs model training across hardware resources.
Using KV Caching, Semantic Caching, Distributed Inference, Quantization, Low-latency API Design, Scaling LLM Serving on GPU and CPU to optimize SLMs (Small Language Models) inference speed, scalability, and resource utilization.
Employing Snowflake, Databricks, and Sagemaker to manage data and ML model training.

Responsibilities

Engage in cutting-edge initiatives to enhance virtual assistant capabilities.
Leverage Natural Language Processing (NLP), deep learning, and generative AI techniques.
Drive the development of scalable, production-grade AI solutions through experimentation with LLMs and SLMs.
Optimize SLMs using advanced model fine-tuning techniques.
Lead brainstorming sessions focused on NLP advancements, LLM fine-tuning strategies, and production deployment.
Support tech and product teams with expertise and guidance in machine learning applications.
Perform monthly releases and model optimization, contributing to the release cycle.
Own the machine learning solution for the transaction search application, including stakeholder management, continuous optimization, and research.
Lead the build of a question-and-answer solution using advanced NLP techniques for the Chase.com website.
Conduct thorough analysis of business needs, exploring state-of-the-art research papers, deep learning models, and generative AI methods.
Design and execute experiments using LLMs and SLMs to enhance performance on targeted tasks.
Utilize adapted models for the finance domain and optimize training efficiency.
Lead AI Solution Development, stakeholder management, model release management, and research on NLP Solutions.