Lead Software Engineer - AI/ML - AI Agent Platform

Fairygodboss; Jersey City (NJ); On-site; USD 180,000 - 260,000; Full time

Job summary

JPMorganChase is seeking a senior engineer to lead the AI Agent Platform, shaping SDKs, orchestration frameworks, and reusable components that empower teams across the firm to build agent-based experiences.

You will own core platform services, drive reliability, performance, and security, and mentor engineers while partnering with product managers to deliver measurable business impact on NL2SQL, RAG, and document-processing capabilities.

Qualifications

• 5+ years of applied AI/ML experience with formal training.
• Advanced Python and strong software engineering fundamentals.
• Experience deploying ML/LLM systems to production.
• Prompt engineering and retrieval-augmented generation knowledge.
• Designing reliable services with incident response readiness.
• Ability to lead technical decisions and mentor others.
• Strong cross-functional communication skills.

Responsibilities

• Deliver AI Agent Platform SDK and frameworks from design to production deployment.
• Own services like orchestration, tool calling, RAG pipelines, and model-serving integration.
• Build agent modalities: voice agents, document-extraction, and agent memory.
• Improve NL2SQL, RAG, voice, and document-extraction capabilities with benchmarks.
• Establish engineering standards and mentorship for reliability and latency.
• Develop testing, observability, and governance capabilities for enterprise use.
• Collaborate with product managers to shape roadmap and success metrics.
• Ensure security, stability, and scalability across teams.
• Contribute to documentation, SDK references, and enablement content.

Skills

Python

AI/ML concepts

ML in production

Prompt engineering

System design

Technical leadership

Communication

Education

AI/ML certification

Tools

Docker

Kubernetes

LangGraph

LlamaIndex

Google ADK

Job description

Help shape how teams across the firm build AI agents. You will lead engineering for a flagship, high-visibility AI Agent Platform - the SDK, orchestration frameworks, and reusable components that power agentic AI experiences across the organization. Beyond any single chat experience, you will own the foundational tooling that lets teams compose agents-conversational, voice, and document-extraction-ground them in enterprise data, give them durable agentic memory, and continuously improve accuracy on hard capabilities like NL2SQL and RAG. You will raise delivery standards, turn complex problems into reliable production systems, and set the quality bar for infrastructure that many teams and thousands of users depend on. Join a team where strong engineering, thoughtful collaboration, and continuous learning are core to how we operate.

Job responsibilities

• Lead end-to-end delivery of the AI Agent Platform SDK and frameworks-from problem framing and technical design through production deployment, scaling, and monitoring-that engineering teams use to build, evaluate, and operate AI agents.
• Own core platform services and reusable building blocks: agent orchestration, tool/function calling, retrieval-augmented generation pipelines, and model-serving integration.
• Build and operate first-class agent modalities on the platform, including voice agents (speech-to-text, text-to-speech, low-latency streaming, and turn-taking), document-extraction agents (parsing, OCR, structured field and table extraction from complex documents), and agentic memory (short- and long-term memory, persistence, retrieval, and context management across sessions).
• Drive systematic hill-climbing of core agent capabilities-including NL2SQL, RAG, voice, document extraction, and tool use-by building evaluation datasets, benchmarks, and quality metrics, then iterating on prompts, retrieval, and orchestration to measurably improve accuracy.
• Establish engineering standards through hands-on system design, rigorous code review, and mentorship to improve reliability, maintainability, latency, and developer experience for teams building on the SDK.
• Build and operationalize evaluation, testing, and observability capabilities (offline eval harnesses, regression suites, tracing, latency and cost telemetry, logs, and analytics) to continuously improve solution quality.
• Implement robust safety and governance patterns-guardrails, content filtering, prompt-injection defenses, access controls, and audit-ready operational practices aligned to enterprise expectations.
• Partner with product managers and stakeholders to shape the roadmap, define success metrics (capability accuracy, adoption, task completion, quality), and prioritize work that delivers measurable business impact.
• Drive cross-functional alignment across engineering, data, security, and risk partners to ensure the platform is secure, stable, performant, and scalable.
• Contribute to technical documentation, SDK references, reference implementations, and enablement content that accelerates adoption and responsible usage.

Required qualifications, capabilities and skills

• Formal training or certification on applied artificial intelligence and machine learning concepts and 5+ years applied experience.
• Advanced proficiency in Python with strong software engineering fundamentals, including testing, design patterns, version control, and code review practices.
• Hands-on experience building, evaluating, and deploying machine learning or large-language-model-enabled systems into production environments-ideally reusable libraries, SDKs, or platform services consumed by other engineering teams.
• Practical experience with prompt engineering and retrieval-augmented generation, including building evaluation methods, benchmarks, and quality measurement to systematically improve capabilities such as NL2SQL and RAG.
• Experience designing and operating reliable services, including incident response readiness, performance tuning, and operational stability for data-intensive systems.
• Demonstrated ability to lead technical decisions and deliver outcomes through ambiguity, balancing speed, risk, and long-term maintainability.
• Strong communication skills with the ability to explain technical trade-offs to both technical and non-technical stakeholders.

Preferred qualifications, capabilities and skills

• Experience with agent orchestration frameworks (for example, LangGraph, LlamaIndex, Google ADK, or custom orchestration) and evaluation tooling for LLM systems.
• Experience building voice agents (speech recognition, text-to-speech, real-time streaming audio, and conversational turn-taking).
• Experience building document-extraction / intelligent document-processing agents (OCR, layout parsing, structured field and table extraction from complex or unstructured documents).
• Experience implementing agentic memory systems - short- and long-term memory, persistence, and context retrieval across sessions.
• Familiarity with vector databases, embedding pipelines, or graph-based memory approaches used in retrieval-augmented generation solutions.
• Experience with continuous integration and continuous delivery practices and containerization (Docker and Kubernetes) for production deployments.
• Experience with cloud and machine learning platforms (for example, Amazon Web Services, Databricks, or comparable platforms).

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.

ABOUT US

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

ABOUT THE TEAM

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.