Your browser does not support javascript! Please enable it, otherwise web will not work for you.

Lead Software Engineer and ML

Home > Python programming jobs

Lead Software Engineer and ML in USA

  • JPMorgan Chase
  • Full time
  • Email
  • Jersey City, NJ

Responsibilities

  • Design, ship, and operate production agents on NEO from prototype through production.
  • Build production-grade Graph RAG retrieval with knowledge-graph traversal, vector search, chunking, ranking, and grounding strategies.
  • Design episodic and semantic agent memory with recall, summarization, and decay policies.
  • Manage entitlement-, lineage-, and tenant-aware organizational context for agent reasoning.
  • Compose multi-agent workflows using A2A and integrate tools and data through MCP servers, including Bitbucket, Confluence, Databricks, Kubernetes, Snowflake, and Splunk.
  • Build task-level and end-to-end evaluations, regression suites, LLM-as-judge workflows, and quality and safety gates.
  • Deploy and operate solutions on AWS and/or Azure using strong SDLC, security, resiliency, and observability practices.
  • Partner with product and business stakeholders across Commercial and Investment Banking and Payments to deliver supported agents.
  • Lead evaluation sessions with external vendors, startups, and internal teams to assess architectures and technical applicability.

Requirements

  • Formal training or certification in software engineering concepts and at least 5 years of applied experience, including at least 4 years working on AI/ML solutions.
  • Strong Python programming skills and advanced knowledge of data structures, algorithms, machine learning, data mining, information retrieval, and statistics.
  • Hands-on experience delivering system design, application development, testing, operational stability, and production LLM-powered or agentic applications.
  • Deep proficiency with Kubernetes and Amazon EKS, micro-VM isolation such as Firecracker, Kata Containers, and gVisor, and sidecar architectures.
  • Experience designing defense in depth across networking and mTLS, workload identity, fine-grained authorization, runtime isolation, and application-level guardrails.
  • Practical RAG experience with retrieval quality, embeddings, and vector stores, plus expert knowledge of AWS, Azure, or Kubernetes.
  • Knowledge of data management and data model design, including real-time processing with SQL and NoSQL stores such as Postgres, OpenSearch, and Redis.
  • Preferred: MS in Computer Science, Statistics, Mathematics, Machine Learning, or a related field, or equivalent experience.
  • Preferred: experience with agent frameworks or runtimes, A2A, MCP, agent memory, organizational context management, knowledge graphs, and Graph RAG.
  • Preferred: knowledge of LLM fine-tuning, small language model inference, modern JavaScript/TypeScript frameworks, financial or payments technology, and Go or Rust.
  • Strong communication skills and ability to partner with senior technical and business stakeholders.

Benefits

  • Competitive total rewards and role-based base salary, with eligible roles potentially receiving commission-based pay or discretionary incentive compensation.
  • Benefits may include comprehensive health care, on-site wellness centers, retirement savings, backup childcare, tuition reimbursement, mental health support, and financial coaching.
  • JPMorganChase is an equal opportunity employer and provides reasonable accommodations.

Salary: $157k - $215k/yr

JPMorgan Chase

With a history tracing its roots to 1799 in New York City, JPMorganChase is one of the world's oldest, largest, and best-known financial institutions—carrying forth the innovative spirit of our heritage firms in global operations across 100 markets. We serve millions of customers and many of the ...

Similar positions

Security Platform Automation Engineer

  • Apple
  • Full time
  • USA
  • 10/15/2026
  • Seattle, WA

Analytics Engineer

  • Reddit
  • Full time
  • USA
  • 10/15/2026
  • Salary: $164k-$230k
  • Remote

Senior AI Engineer

  • Infiterra S.A
  • Full time
  • Remote
  • 10/15/2026
  • Greece, UK

Security Operations Engineer (SecOps)

  • Collibra
  • Full time
  • USA
  • 10/15/2026
  • Salary: $116k - $145k/yr
  • Remote

Accounting Automation Engineer

  • Ruby Labs
  • Full time
  • Remote
  • 10/15/2026
  • EMEA