Applied AI ML Lead - AI Agents & Agentic Systems

JPMorgan Chase & Co.; Palo Alto (CA); On-site; USD 210,000 - 260,000; Full time

Job summary

JPMorgan Chase & Co. seeks an Applied AI and Machine Learning Lead to drive the design and delivery of ML and deep learning solutions at scale within the AI and Machine Learning and Data Platform team in Corporate Sector.

You will own problem framing, experimentation, and productionization, mentoring teams and raising engineering standards. The role focuses on applying a wide range of ML techniques, evaluating new AI methods, and delivering robust, scalable solutions that meet business needs.

Qualifications

• MS/PhD in CS/ML or related field with 5+ years of applied ML experience.
• Strong programming skills in Python; Java/C/C++ proficiency desirable.
• Deep knowledge of ML, LLMs, and agentic techniques; productionization of models.

Responsibilities

• Serve as SME on ML techniques and optimizations across production workloads.
• Lead end-to-end design, experimentation, and productionization of models.
• Mentor engineers and promote robust engineering practices.
• Collaborate with product and data teams to deliver scalable ML solutions.
• Integrate Generative AI within ML Platform and drive architectural decisions.

Skills

Python

Java/C/C++

LLMs

PyTorch/TensorFlow

GPU optimization

Distributed training

Excellent communication

Education

MS or PhD in Computer Science / Machine Learning or related field

Tools

SageMaker

AWS

EKS

PyTorch

TensorFlow

Job description

Join a team where your work directly shapes how machine learning is applied at scale across the firm. You'll partner with product, engineering, and data teams to take ideas from experimentation through production, improving outcomes through thoughtful model development, evaluation, and operational excellence.

Applied AI and Machine Learning Lead at JPMorganChasewithin the AI and Machine Learning and Data Platform team in Corporate Sector, you will drive the design and delivery of machine learning and deep learning solutions that solve meaningful business problems. You will take ownership from problem framing and experimentation through productionization, ensuring solutions are robust, scalable, and measurable. You will also help raise the technical bar through mentorship, strong engineering practices, and a culture of continuous learning.

Job responsibilities

• Serve as a subject matter expert on a wide range of ML techniques and optimizations.
• Provide in-depth knowledge of ML algorithms, frameworks, and techniques.
• Enhance ML workflows through advanced proficiency in large language models (LLMs) and related techniques.
• Conduct experiments to evaluate and benchmark latest AI and agentic techniques, analyzing results, tuning models and agentic systems.
• Hands on coding to bring the experimental results into production solutions by collaborating with engineering team. Owning end to end code development in python for both proof of concept/experimentation and production-ready solutions.
• Optimizing system accuracy and performance by identifying and resolving inefficiencies and bottlenecks. Collaborates with product and engineering teams to deliver tailored, science and technology-driven solutions.
• Integrate Generative AI within the ML Platform using state-of-the-art techniques.
• Drives decisions that influence the product design, application functionality, and technical operations and processes.

Required qualifications, capabilities, and skills

• MS and/or PhD in Computer Science, Machine Learning, or a related field, with at least 5 years of applied machine learning experience.
• At least 5 year's experience in one of the programming languages like Python, Java, C/C++, etc. Intermediate Python is a must.
• At least 5 years' experience in applying data science, ML techniques to solve business problems.
• Solid background in Large Language Models (LLMs) and agentic applied science such as mutli-agent orchestration, reasoning, skills, tools.
• Experience with applied research and experimentation on machine learning and deep learning methods (including LLMs/GenAI).
• Deep understanding and expertise in deep learning frameworks such as PyTorch or TensorFlow.
• Experience in advanced applied ML areas such as GPU optimization, finetuning, embedding models, inferencing, prompt engineering, evaluation, RAG (Similarity Search), reasoning, context management, and other advanced agentic capabilities.
• Ability to work on tasks and projects through to completion with limited supervision.
• Passion for detail and follow through. Excellent communication skills and team player
• Demonstrated leadership in working effectively with engineers, product managers, and other ML practitioners.

Preferred qualifications, capabilities, and skills

• Experience withdistributed training frameworks.
• In-depth understanding of advanced methodologies such as Search/Ranking, Recommender systems, Graph techniques, multi-agent orchestration, evaluation, benchmarking.
• Advanced knowledge in Reinforcement Learning or Meta Learning.
• Deep understanding of Large Language Model (LLM) techniques, including Agents, Planning, Reasoning, and other related methods.
• Experience with building and deploying ML models on cloud platforms such as AWS and AWS tools like Sagemaker, EKS, etc.

FEDERAL DEPOSIT INSURANCE ACT: This position is subject to Section 19 of the Federal Deposit Insurance Act. As such, an employment offer for this position is contingent on JPMorganChase's review of criminal conviction history, including pretrial diversions or program entries.