Principal Software Engineer - AI Engineer

Jersey City is calling for a seasoned AI engineering leader to guide a high-impact program within JPMorganChase. The Principal Software Engineer - AI Engineer will drive the full ML lifecycle, craft agentic AI architectures, and deliver scalable, LLM-enabled solutions, while mentoring teams and establishing engineering standards across the organization.

Responsibilities

• Design and implement agentic AI reference architectures, including orchestration, retrieval, memory, guardrails, and evaluation harnesses.
• Write production-quality Python code (PyTorch or TensorFlow as needed) and review critical-path code.
• Create reusable components for prompt management, evaluators, safety filters, connectors, embeddings pipelines, and memory stores.
• Build and operate LLM-powered APIs and microservices integrated into advisor, client, and internal workflows.
• Own the end-to-end ML lifecycle: experimentation, CI/CD, automated testing, monitoring, drift detection, versioning, and rollback.
• Optimize inference for latency, throughput, caching, batching, model selection, and cost per inference.
• Partner with data teams on structured and unstructured data pipelines, document ingestion, metadata, and access controls.
• Set engineering standards for agentic AI systems and lead design reviews.
• Influence roadmap and priorities through technical insight and delivery.
• Architect and govern agentic AI enabled engineering workflows using enterprise-approved tools to improve delivery speed, code quality, and operational outcomes at scale, including AI-driven PR review assistance, test generation/maintenance, release readiness checks, incident triage and root-cause acceleration, while defining guardrails for validation, security, resiliency, and reuse across teams.
• Apply knowledge of tools within the Software Development Life Cycle toolchain, including enterprise-approved AI-assisted development and automation capabilities, to enhance automation value at scale.

Requirements

• Formal training or certification in software engineering concepts plus 7+ years of applied experience.
• Strong Python engineering skills with experience in PyTorch or TensorFlow.
• Expertise with vector storage systems and designing memory for agents.
• Experience developing long-running agents that operate autonomously using tools, skills, and human in the loop.
• Proven track record deploying LLM-backed services to production (APIs, microservices).
• Deep MLOps experience, including CI/CD, monitoring, incident response, and model governance.
• Cloud-native AI deployment experience (AWS or Azure) with cost and performance optimization.
• Commitment to responsible AI practices and operational excellence.
• Strong communication and collaboration skills across product, risk, legal, and compliance teams.
• Experience designing and leading adoption of agentic AI enabled development practices across teams, including standards for human-in-the-loop validation, auditability/traceability of changes, and secure handling of sensitive data.
• Understanding of responsible AI use and governance expectations in engineering workflows; ability to influence senior technical leaders on safe scaling patterns and reuse.

Technologies

• Python
• PyTorch
• TensorFlow
• Vector storage systems

Benefits

• Base salary determined by role, experience, skill set and location
• Commission-based pay and/or discretionary incentive compensation, paid in cash and/or forfeitable equity
• Comprehensive health care coverage
• On-site health and wellness centers
• Retirement savings plan
• Backup childcare
• Tuition reimbursement
• Mental health support
• Financial coaching

Federal Deposit Insurance Act

This position is subject to Section 19 of the Federal Deposit Insurance Act. As such, an employment offer is contingent on JPMorganChase's review of criminal conviction history, including pretrial diversions or program entries.

About the Team

Our Corporate Functions teams span finance, risk, human resources, and marketing, forming a vital backbone for the company. They ensure that we set our businesses, clients, customers, and employees up for success by supporting operations with integrity and compliance.