Lead Software Engineer - AI/ML Engineering, GPU ML Serving

Aumni; Palo Alto (CA); On-site; USD 170,000 - 260,000; Full time

Benefits offered by this job

Health insurance

Retirement plan

Tuition reimbursement

Mental health support

Equity

Job summary

JPMorgan Chase is seeking a Software Engineer III to join the Commercial and Investment Banking team. You will design, build, and secure scalable technology products while applying ML and AI-assisted development tools to accelerate delivery and quality.

You will mentor peers, contribute to architecture decisions, and ensure reliable, performant services across distributed systems and APIs, leveraging Python and cloud-native technologies in a security-conscious environment.

Qualifications

• Formal training or certification on software engineering concepts and 5+ years applied experience.
• Professional software development experience, with emphasis on ML systems.
• Hands-on experience using enterprise-authorized AI-assisted software development tools within the work environment with demonstrated ability to critically evaluate, validate, and refine AI-generated outputs for correctness, performance, and security.
• Responsible for AI use in engineering workflows, including data sensitivity considerations, secure handling of inputs/outputs, and adherence to resiliency and security expectations; ability to guide peers on safe and effective usage within team practices.
• Experience contributing to the design and development of scalable systems and services (e.g., web services, APIs)
• Proficiency in Python, including experience with ML frameworks (e.g., TensorFlow, PyTorch, or similar)
• Experience with cloud-native technologies (e.g., Docker, Kubernetes)
• Experience with one or more public cloud platforms (AWS, GCP)
• Experience with system design fundamentals (e.g., reliability, scalability, performance tradeoffs)
• Experience working with distributed systems concepts, microservices architectures, Cassandra or equivalent NoSQL databases
• Mentorship and technical leadership experience, including setting engineering standards and elevating overall team quality

Responsibilities

• Executes software solutions, design, development, and technical troubleshooting with ability to think beyond routine or conventional approaches to build solutions or breakdown technical problems
• Creates secure and high-quality production code and maintains algorithms that run synchronously with appropriate systems
• Leverages enterprise-authorized AI coding assist tools within the work environment to improve code quality, delivery speed, and productivity across complex deliverables (e.g., code generation/refactoring, unit test creation, documentation), while validating outputs through peer review, automated testing, and secure coding standards; contributes learnings and reusable patterns to improve broader team effectiveness.
• 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.
• Produces architecture and design artifacts for complex applications while being accountable for ensuring design constraints are met by software code development
• Gathers, analyzes, synthesizes, and develops visualizations and reporting from large, diverse data sets in service of continuous improvement of software applications and systems
• Proactively identifies hidden problems and patterns in data and uses these insights to drive improvements to coding hygiene and system architecture

Skills

ML systems

Python

Cloud-native tech

API design

Secure coding

AI-assisted development

Distributed systems

Mentorship

Tools

Docker

Kubernetes

TensorFlow/PyTorch

AI coding tools

Job description

We have an exciting and rewarding opportunity for you to take your software engineering career to the next level.

As a Software Engineer III at JPMorgan Chase within the Commercial and Investment Banking team, you will be a key member of an agile team, designing and delivering secure, stable, and scalable technology products. You will implement critical technology solutions across multiple technical areas, supporting the firm’s business objectives.

Job responsibilities

• Executes software solutions, design, development, and technical troubleshooting with ability to think beyond routine or conventional approaches to build solutions or breakdown technical problems
• Creates secure and high-quality production code and maintains algorithms that run synchronously with appropriate systems
• Leverages enterprise-authorized AI coding assist tools within the work environment to improve code quality, delivery speed, and productivity across complex deliverables (e.g., code generation/refactoring, unit test creation, documentation), while validating outputs through peer review, automated testing, and secure coding standards; contributes learnings and reusable patterns to improve broader team effectiveness.
• 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.
• Produces architecture and design artifacts for complex applications while being accountable for ensuring design constraints are met by software code development
• Gathers, analyzes, synthesizes, and develops visualizations and reporting from large, diverse data sets in service of continuous improvement of software applications and systems
• Proactively identifies hidden problems and patterns in data and uses these insights to drive improvements to coding hygiene and system architecture

Required qualifications, capabilities, and skills

• Formal training or certification on software engineering concepts and 5+ years applied experience
• Professional software development experience, with emphasis on ML systems.
• Hands-on experience using enterprise-authorized AI-assisted software development tools within the work environment (e.g., for coding, test creation, troubleshooting, or documentation) with demonstrated ability to critically evaluate, validate, and refine AI-generated outputs for correctness, performance, and security.
• Responsible for AI use in engineering workflows, including data sensitivity considerations, secure handling of inputs/outputs, and adherence to resiliency and security expectations; ability to guide peers on safe and effective usage within team practices.
• Experience contributing to the design and development of scalable systems and services (e.g., web services, APIs)
• Proficiency in Python, including experience with ML frameworks (e.g., TensorFlow, PyTorch, or similar)
• Experience with cloud-native technologies (e.g., Docker, Kubernetes)
• Experience with one or more public cloud platforms (AWS, GCP)
• Experience with system design fundamentals (e.g., reliability, scalability, performance tradeoffs)
• Experience working with distributed systems concepts, microservices architectures, Cassandra or equivalent NoSQL databases
• Mentorship and technical leadership experience, including setting engineering standards and elevating overall team quality

Preferred qualifications, capabilities, and skills

• Working knowledge of additional languages such as Java and/or C++
• Exposure to deploying or operating GPU workloads in Kubernetes environments
• Exposure to ML model serving frameworks (e.g., TorchServe, TensorFlow Serving, Triton Inference Server)
• Familiarity with distributed training and inference systems
• Awareness of model compression techniques and edge deployment concepts
• Familiarity with API management concepts, Domain-Driven Design (DDD) and working in Agile development environments

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 JPMorgan Chase’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 the FAQs for more information about requesting an accommodation.

JPMorgan Chase & Co. is an Equal Opportunity Employer, including Disability/Veterans

J.P. Morgan’s Commercial & Investment Bank is a global leader across banking, markets, securities services and payments. Corporations, governments and institutions throughout the world entrust us with their business in more than 100 countries. The Commercial & Investment Bank provides strategic advice, raises capital, manages risk and extends liquidity in markets around the world.

Lead Software Engineer - AI/ML Deep Learning & GPU ML Serving