Applied AI/ML Lead: Scale ML from Idea to Production

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

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.