Applied AI & ML Lead — Production-Grade, Scalable Solutions

JPMorganChase; Palo Alto (CA); On-site; USD 180,000 - 240,000; Full time

Job summary

JPMorganChase seeks an Applied AI and Machine Learning Lead to shape machine learning solutions at scale within the AI and ML Data Platform team in Corporate Sector. You will own problem framing, experimentation, deployment, and continuous improvement, mentoring engineers and aligning with product goals.

Your role emphasizes robust, scalable ML/DL systems, with hands-on Python development and expertise in LLMs, agentic frameworks, and GPU optimization to drive measurable business impact.

Qualifications

• 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 years' 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 multi-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.

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.

Skills

Python

Java

C/C++

ML techniques

LLMs

Team leadership

Communication

Problem solving

Education

MS/PhD in CS or related field

Tools

PyTorch

TensorFlow

SageMaker

EKS

Job description

JPMorganChase seeks an Applied AI and Machine Learning Lead to shape machine learning solutions at scale within the AI and ML Data Platform team in Corporate Sector. You will own problem framing, experimentation, deployment, and continuous improvement, mentoring engineers and aligning with product goals.

Your role emphasizes robust, scalable ML/DL systems, with hands-on Python development and expertise in LLMs, agentic frameworks, and GPU optimization to drive measurable business impact.