Principal Software Engineer - AI Engineer

Next Frontier Capital; Palo Alto (CA); On-site; USD 210,000 - 290,000; Full time

Job summary

JPMorganChase seeks a Principal Software Engineer in AI/ML & Data Platforms for LLM Suite, leading a technical area across teams and projects. You will mentor engineers and drive scalable ML solutions with operational excellence.

You will design agentic AI architectures, implement production-grade code, build LLM APIs, and own the full ML lifecycle from experimentation to deployment, ensuring security and governance at scale.

Qualifications

• Formal training or certification on software engineering concepts and 7+ years applied experience.
• Strong Python engineering skills with PyTorch or TensorFlow experience.
• Expertise with vector storage and memory design for agents.
• Experience deploying LLM-backed services to production (APIs, microservices).
• Deep MLOps experience: CI/CD, monitoring, incident response, governance.
• Cloud-native AI deployment (AWS or Azure) with cost/performance optimization.
• Commitment to responsible AI practices and operational excellence.
• Strong cross-functional collaboration with product, risk, legal and compliance.
• Experience leading adoption of agentic AI development practices and human-in-the-loop validation.

Responsibilities

• Design and implement agentic AI architectures including orchestration, memory, guardrails and evaluation harnesses.
• Write production-quality Python code (PyTorch or TensorFlow) and review critical-path code.
• Create reusable components for prompt management, evaluators, safety filters, connectors and embeddings pipelines.
• 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 and rollback.
• Optimize inference for latency, throughput, caching, batching, model selection and cost per inference.
• Partner with data teams on 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 with guardrails for security and reuse.

Skills

Python engineering

PyTorch

TensorFlow

ML engineering

MLOps

APIs & microservices

Cloud deployment

Agent memory design

CI/CD

Security & governance

Job description

If you are looking for a game-changing career, working for one of the world's leading financial institutions, you've come to the right place.

As a Principal Software Engineer at JPMorganChase within the Corporate Sector – AI/ML & Data Platforms for LLM Suite, you will lead a specialized technical area, driving impact across teams, technologies, and projects. In this role, you will leverage your deep knowledge of machine learning, software engineering, and product management to spearhead multiple complex ML projects and initiatives, serving as the primary decision-maker and a catalyst for innovation and solution delivery. LLM Suite is JPMorganChase’s premier internally built AI tool leveraged by +250k employees for everything from individual productivity to larger scale, business solutions.

You will be responsible for hiring, leading, and mentoring a team of Machine Learning and Software Engineers, focusing on best practices in ML engineering, with the goal of elevating team performance to produce high-quality, scalable ML solutions with operational excellence.

Job 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
• Architects and governs agentic AI-enabled engineering workflows (using enterprise-authorized tools within the work environment) to improve delivery speed, code quality, and operational outcomes at scale (e.g., 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.
• Applies knowledge of tools within the Software Development Life Cycle toolchain, including enterprise-authorized AI-assisted development and automation capabilities, to improve the value realized by automation at scale.

Required Qualifications, Capabilities, and Skills

• Formal training or certification on software engineering concepts and 7+ years applied experience
• Strong Python engineering skills; experience with PyTorch or TensorFlow
• Expertise working with Vector storage systems and designing memory for Agents
• Expertise developing long running agents that run autonomously using tools, skills and human in the loop
• Proven experience 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
• Demonstrated commitment to responsible AI practices and operational excellence
• Strong communication and collaboration skills, working across product, risk, legal, and compliance teams
• Demonstrated experience designing and leading adoption of agentic AI-enabled development practices (using enterprise-authorized tools within the work environment) across teams, including setting standards for human-in-the-loop validation, auditability/traceability of changes, and secure handling of sensitive data.
• Strong understanding of responsible AI use and control expectations in engineering workflows, including security/resiliency implications, data sensitivity, and risk-based governance; ability to influence senior technical leaders on safe scaling patterns and reuse.

Preferred Qualifications, Capabilities, and Skills

• Experience with fine-tuning, adapters, or custom evaluation frameworks
• Background operating AI systems in regulated environments (finance, healthcare, etc.)
• Experience with prompt engineering and LLM orchestration
• Knowledge of safety filters, audit logging, and explainability in production systems
• Experience mentoring senior engineers and leading architecture discussions
• Demonstrated ability to influence technical roadmaps and priorities

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.

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

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. Provide expertise and engineering excellence to enhance, build and deliver market-leading technologies within the firm