Applied AI ML Lead - AI Agents & Agentic Systems

J.P. Morgan; New York (NY); On-site; USD 180,000 - 240,000; Full time

Job summary

JPMorganChase seeks an Applied AI and Machine Learning Lead to design and deliver machine learning solutions across the firm. You will own project framing, experimentation, and productionization while ensuring robust, scalable results and measurable impact.

You will mentor engineers and collaborate with product, engineering, and data teams to raise the technical bar and drive continuous learning across the organization.

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 year's experience in 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 NLP and Large Language Models (LLMs).
• Experience with ML and deep learning methods.
• Deep learning frameworks such as PyTorch or TensorFlow.
• Experience in advanced applied ML areas like GPU optimization, finetuning, embedding models, inference, prompt engineering, evaluation, and RAG.

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 with proficiency in LLMs and related techniques.
• Conduct experiments with latest ML technologies, analyze results, and tune models.
• Hands-on coding to production with Python for proof of concept and production-ready solutions.
• Optimize system accuracy and performance by identifying bottlenecks and inefficiencies.
• Integrate Generative AI within the ML Platform using state-of-the-art techniques.
• Drive decisions that influence product design and technical operations.

Skills

Python

C/C++

NLP

LLMs

PyTorch

TensorFlow

Deployment

Leadership

Education

MS/PhD in CS or ML

Tools

Ray

MLFlow

AWS SageMaker

GPU optimization

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.

As an Applied AI and Machine Learning Lead at JPMorganChase within 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.
• Conducting experiments using latest ML technologies, analyzing results, tuning models
• 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 Natural Language Processing (NLP) and Large Language Models (LLMs)
• Experience with machine learning and deep learning methods.
• 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).
• 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 with Ray, MLFlow, and/or other distributed training frameworks.
• In-depth understanding of Search/Ranking, Recommender systems, Graph techniques, and other advanced methodologies.
• 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.

#LI-RB3

#AMDPAIML

JPMorganChase, one of the oldest financial institutions, offers innovative financial solutions to millions of consumers, small businesses and many of the world’s most prominent corporate, institutional and government clients under the J.P. Morgan and Chase brands. Our history spans over 200 years and today we are a leader in investment banking, consumer and small business banking, commercial banking, financial transaction processing and asset management.

We offer a competitive total rewards package including base salary determined based on the role, experience, skill set and location. Those in eligible roles may receive commission-based pay and/or discretionary incentive compensation, paid in the form of cash and/or forfeitable equity, awarded in recognition of individual achievements and contributions. We also offer a range of benefits and programs to meet employee needs, based on eligibility. These benefits include comprehensive health care coverage, on-site health and wellness centers, a retirement savings plan, backup childcare, tuition reimbursement, mental health support, financial coaching and more. Additional details about total compensation and benefits will be provided during the hiring process.

We recognize that our people are our strength and the diverse talents they bring to our global workforce are directly linked to our success. We are an equal opportunity employer and place a high value on diversity and inclusion at our company. We do not discriminate on the basis of any protected attribute, including race, religion, color, national origin, gender, sexual orientation, gender identity, gender expression, age, marital or veteran status, pregnancy or disability, or any other basis protected under applicable law. We also make reasonable accommodations for applicants’ and employees’ religious practices and beliefs, as well as mental health or physical disability needs. Visit our FAQs for more information about requesting an accommodation.

JPMorgan Chase & Co. is an Equal Opportunity Employer, including Disability/Veterans

Our professionals in our Corporate Functions cover a diverse range of areas from finance and risk to human resources and marketing. Our corporate teams are an essential part of our company, ensuring that we’re setting our businesses, clients, customers and employees up for success. Lead applied AI and machine learning solutions from experimentation to production to deliver measurable business impact.