Lead Machine Learning Engineer - Generative AI and Agent Platforms

JPMorgan Chase offers an onsite opportunity in Jersey City, NJ for a Lead Machine Learning Engineer focused on Generative AI and Agent Platforms. The role provides a competitive annual base salary in the range of USD 164,350 to 260,000, along with a comprehensive benefits package designed to support health, finances, learning, and work-life balance. Expect health care coverage, on-site health and wellness centers, a retirement savings plan, backup childcare, tuition reimbursement, mental health support, financial coaching, and the potential for commission-based pay or discretionary incentive compensation (cash and/or forfeitable equity).

In this role you will lead the design, build, and operation of agent platforms and AI agents for payments technology, applying MLOps and enterprise-scale capabilities across multiple business units.

Responsibilities

• Lead end-to-end development of production AI agents across a federated portfolio, from prototype to launch and ongoing operational support.
• Build robust production retrieval systems, implementing hybrid retrieval-augmented generation (vector search plus graph-based retrieval when appropriate) with solid chunking, ranking, and grounding.
• Develop agent memory patterns, including episodic and semantic memory, with recall, summarization, and decay policies aligned to use-case needs.
• Create entitlement- and tenant-aware context assembly so agents reason only over permitted data, enabling traceability and auditability.
• Coordinate multi-agent workflows and integrate external tools and data sources through secure connectors and standardized tool interfaces.
• Develop evaluation frameworks, including task-level and end-to-end assessments, regression suites, automated scoring, and release gates for quality and safety.
• Deploy and operate agent services on AWS and/or Azure, following strong software development lifecycle practices, with emphasis on security, resiliency, and observability.
• Improve runtime performance and reliability by instrumenting tracing, monitoring, and incident-response playbooks for agent services.
• Partner with product and business leaders to translate use cases into shipped capabilities, define success metrics, and drive continuous improvement from production feedback.

Requirements

• Formal training or certification in applied AI and machine learning concepts with at least 5 years of applied experience.
• Bachelor’s degree in Computer Science, Engineering, Statistics, Mathematics, or a related field, or equivalent practical experience.
• Minimum 7 years of software development experience, including at least 4 years delivering AI or ML solutions.
• Hands-on production experience building large language model powered or agentic applications, including tracing, evaluations, and safety guardrails.
• Strong Python programming skills with solid fundamentals in data structures, algorithms, and applied statistics.
• Practical experience with retrieval-augmented generation, including embedding strategies, retrieval quality measurement, and use of vector databases.
• Experience operating production workloads on AWS, Azure, or Kubernetes.
• Experience designing data models and building systems using SQL and NoSQL technologies for real-time or near-real-time use cases.
• Strong communication skills and the ability to partner effectively with senior technical and business stakeholders.

Technologies

• Databricks
• GenAI Gateway
• Amazon Web Services
• Microsoft Azure
• Kubernetes
• Python
• Vector databases
• Go
• Rust
• JavaScript
• TypeScript
• Graph databases
• SQL
• NoSQL

About JPMorgan Chase

JPMorgan Chase is one of the oldest financial institutions, serving millions of consumers, small businesses, and many of the world’s leading corporate, institutional, and government clients under the J. P. Morgan and Chase brands. With a history spanning more than 200 years, the firm is a leader across investment banking, consumer and small business banking, commercial banking, financial transaction processing, and asset management.

Compensation and inclusion

We offer a competitive total rewards package, with base salary determined by the role, experience, skill set, and location. Eligible roles may include commission-based pay and/or discretionary incentive compensation (cash and/or forfeitable equity). JPMorgan Chase is an equal opportunity employer that values diversity and inclusion and does not discriminate on protected attributes.

About the team

JPMorgan Chase's Commercial and Investment Bank operates as a global leader across banking, markets, securities services, and payments. Corporations, governments, and institutions around the world rely on our expertise in more than 100 countries for strategic advice, capital formation, risk management, and liquidity solutions.