Applied AI ML Lead - Machine Learning Engineer - Agentic Commerce

Job description

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