Applied AI ML Lead - Machine Learning Engineer - Agentic Commerce

Join us to shape the future of AI-powered solutions at JPMorganChase. You’ll leverage the firm’s scale, data, and technology to deliver measurable impact across the Commercial & Investment Bank and Payments. As a Lead AI and ML Engineer, you’ll collaborate with talented teams in a fast-paced environment, building agents that real businesses depend on. We offer opportunities for career growth, exposure to cutting-edge platforms, and the chance to make a difference in a regulated, secure setting.

As a Lead AI and ML Engineer in Digital & Platform Services / Data Analytics, you will design, productionize, and operate LLM-powered Agentic Commerce B2B agents on NEO. You will apply MLOps for automation, continuous delivery, and compliance, turning innovative ideas into shipped, production-grade agents. You’ll partner closely with business, product, data science, and engineering teams, expanding NEO’s portfolio of production agents across CIB sub-LOBs and Payments. Your work will help drive secure, auditable, and impactful AI solutions.

Job Responsibilities:

• Design and ship production agents on NEO, owning them from prototype through production

• Build robust retrieval systems using Graph RAG, knowledge-graph traversal, vector search, chunking, ranking, and grounding strategies

• Design agent memory, including episodic and semantic memory nodes, recall, summarization, and decay policies

• Manage organizational context, assembling entitlement-, lineage-, and tenant-aware context for secure agent reasoning

• Compose multi-agent workflows using A2A and integrate tools and data through MCP servers (Bitbucket, Confluence, Databricks, Kubernetes, Snowflake, Splunk)

• Build and run task-level and end-to-end agent evaluations, regression suites, LLM-as-judge, and quality/safety gating

• Deploy and operate solutions on public cloud (AWS and/or Azure) with strong SDLC, security, resiliency, and observability practices

• Partner with product and business teams to turn use cases into shipped, supported agents

• Build traditional ML model training pipelines and productionize them using MLOps best practices

• Develop batch and online inference for ML models

Required Qualifications, Capabilities, and Skills:

• MS in Computer Science, Statistics, Mathematics, Machine Learning, or related field (or equivalent experience)

• Hands-on experience building LLM-powered or agentic applications in production, including tracing, evaluations, and guardrails

• Strong programming skills in Python, with deep knowledge of data structures, algorithms, machine learning, data mining, information retrieval, and statistics

• Knowledge of Kubernetes (AWS EKS)

• Experience with training models in Databricks and SageMaker

• Experience working with MLFlow

• Practical RAG experience—retrieval quality, embeddings, and vector stores; Graph RAG a strong plus

• Expert knowledge of at least one of: AWS, Azure, Kubernetes

• Knowledge of data management and data model design; real-time processing using SQL (e.g., Postgres) and NoSQL stores (e.g., OpenSearch, Redis)

• Excellent communication skills with the ability to partner effectively with senior technical and business stakeholders

Preferred Qualifications, Capabilities, and Skills:

• Experience with agent frameworks or runtimes, A2A, or MCP

• Agent memory design (memory nodes, episodic/semantic memory) and organizational context management

• Knowledge graphs and graph databases used for retrieval

• Understanding of LLM fine-tuning and small language model inference

• Ability to develop full-stack products using modern JavaScript/TypeScript frameworks (e.g., Next.js, Svelte) for agent UIs (AG-UI / NEO UI SDK)

• Experience working in the financial or payments domain at a large institution