Senior Lead Software Engineer, AI Platforms

Job Description

Be an integral part of an agile engineering team that is constantly pushing the boundaries of what is possible across cloud, infrastructure, and AI/ML platforms. 

As a Senior Lead Software Engineer at JPMorganChase within the Corporate Sector, Infrastructure Platforms team, you will play a critical role in designing, building, and operating secure, scalable, and resilient infrastructure platforms that power enterprise AI/ML workloads. You will help deliver market-leading technology products in a secure, stable, and highly available manner while driving meaningful business impact through deep technical expertise, engineering leadership, and strong problem-solving capabilities.

In this role, you will partner closely with AI/ML engineering teams, platform teams, product owners, and infrastructure stakeholders to translate complex compute, storage, networking, GPU, and scalability requirements into production-ready platforms for multi-GPU and multi-node model training. You will also help advance automation, developer productivity, responsible AI-assisted engineering practices, and operational excellence across the software delivery lifecycle. 

Job Responsibilities 

• Design, develop, test, and deliver secure, high-quality production code; review, debug, and improve code written by others. 
• Architect, build, and operate secure, scalable cloud infrastructure platforms optimized for multi-GPU and multi-node AI/ML training workloads. 
• Partner with AI/ML, data science, and platform engineering teams to translate compute, storage, networking, GPU, and scalability needs into robust infrastructure requirements and platform capabilities. 
• Drive technical design decisions that influence product architecture, application functionality, infrastructure strategy, and operational effectiveness. 
• Monitor, manage, and optimize cloud and GPU infrastructure resources for performance, reliability, utilization, scalability, and cost efficiency. 
• Build and maintain CI/CD pipelines, automation frameworks, and infrastructure-as-code solutions to streamline ML platform deployment, operations, and lifecycle management. 
• Provide technical leadership and guidance to engineers, contractors, and vendor partners, ensuring solutions align with business priorities, engineering standards, security expectations, and long-term platform strategy. 
• Apply deep knowledge of the Software Development Life Cycle toolchain, including enterprise-approved AI-assisted development and automation capabilities, to improve engineering productivity and automation at scale. 
• Champion firmwide SDLC frameworks, engineering standards, secure coding practices, resiliency expectations, and operational best practices. 
• Drives adoption and governance of approved AI-assisted engineering practices across teams to improve code quality, delivery speed, and operational outcomes (e.g., AI-assisted code review/refactoring, test acceleration, release readiness, incident/root-cause analysis), while establishing measurable validation standards (secure coding, peer review, automated testing) and promoting reuse of proven patterns and automation within the SDLC/TLM toolchain.
• Applies knowledge of tools within the Software Development Life Cycle toolchain, including approved 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 5+ years applied experience
• Hands-on experience building highly scalable and highly available infrastructure for machine learning training and/or inference workloads. 
• Solid system-level understanding of GPU infrastructure, accelerators, high-speed interconnects, distributed compute, and related platform technologies. 
• Strong experience with Kubernetes and containerization technologies, including Docker, cluster operations, workload scheduling, observability, and production troubleshooting. 
• Proficiency in at least one modern programming language, such as Python, Go, Java, or C#.
• Demonstrated ability to independently solve complex design, scalability, reliability, performance, and functionality challenges with minimal oversight. 
• Deep understanding of cloud component architecture, including microservices, compute, storage, networking, security, routing, and switching technologies. 
• Demonstrated experience leading effective use of enterprise-authorized AI-assisted software development tools for coding, code review, test acceleration, troubleshooting, and engineering productivity. 
• Demonstrated experience leading effective use of enterprise-authorized AI-assisted software development tools within the work environment (e.g., for coding, code review, test acceleration, troubleshooting) with the ability to set team expectations for validating AI outputs for correctness, performance, and security
• Strong understanding of responsible AI use in engineering workflows, including data sensitivity considerations, secure handling of inputs/outputs, and adherence to resiliency and security expectations; experience coaching senior engineers/leads on compliant usage patterns and controls. 
• Design, develop, test, and deliver secure, high-quality production code; review, debug, and improve code written by others.

 

Preferred Qualifications, Capabilities, and Skills

• Hands-on experience with performance monitoring, production debugging, profiling, sampling, bottleneck analysis, and capacity optimization. 
• Foundational understanding of NVIDIA GPU infrastructure software and ecosystem tooling, such as DCGM, BCM, NVIDIA drivers, CUDA, and training libraries. 
• Experience with MLOps platforms and tooling, including MLflow or similar model lifecycle management solutions. 
• Background in high-performance computing, distributed systems, and ML frameworks, including distributed training, Ray.io, Slurm, or similar workload orchestration technologies. 
• Strong knowledge of network architecture, including high-throughput and low-latency networking patterns for distributed compute and AI/ML workloads. 
• Familiarity with cloud data services, big data processing platforms, Linux systems, and storage technologies used in large-scale data and ML environments. 
• Experience designing platforms that support model training, experiment tracking, feature pipelines, model serving, and scalable inference in enterprise 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

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.