JPMorganChase Logo

JPMorganChase

Senior Lead Software Engineer- AI/ML Training Infrastructure

Reposted Yesterday
Be an Early Applicant
Hybrid
Palo Alto, CA, USA
Senior level
Hybrid
Palo Alto, CA, USA
Senior level
Designs, builds, and operates secure, scalable cloud and GPU infrastructure platforms for enterprise AI/ML workloads. Leads architecture, production coding, Kubernetes and container operations, CI/CD, infrastructure automation, performance optimization, and reliability efforts. Partners with AI/ML and platform teams to support distributed multi-GPU training and inference. Provides technical leadership while advancing responsible AI-assisted engineering, secure SDLC practices, automation, and operational excellence.
The summary above was generated by AI

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 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 TeamOur 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.

JPMorganChase San Francisco, California, USA Office

San Francisco, CA, United States

Similar Jobs

17 Minutes Ago
Hybrid
208K-332K Annually
Expert/Leader
208K-332K Annually
Expert/Leader
AdTech • eCommerce • Information Technology • Software • Travel • Generative AI
Leads Expedia Group’s enterprise leadership development strategy, roadmap, and scalable learning experiences. Integrates leadership development with talent management, succession, performance, employee experience, organizational effectiveness, and culture. Partners with senior leaders, HRBPs, and People teams to diagnose needs, measure outcomes, and deliver practical solutions. Explores responsible AI applications for personalization, coaching, insights, content discovery, and learning effectiveness while guiding leaders in AI-enabled workplaces.
Top Skills: Artificial Intelligence (Ai)Digital Learning TechnologiesPeople Analytics
8 Hours Ago
Remote or Hybrid
USA
19-19 Hourly
Junior
19-19 Hourly
Junior
eCommerce • Legal Tech • Professional Services • Software • Data Privacy
Manage a portfolio of established customer accounts via inbound and outbound calls, uncover upsell and retention opportunities, meet sales and retention targets, monitor purchasing patterns, perform follow-up outreach, and build long-term customer relationships in a metrics-driven environment.
8 Hours Ago
Hybrid
26K-31K Hourly
Junior
26K-31K Hourly
Junior
eCommerce • Legal Tech • Professional Services • Software • Data Privacy
Manage the commercial lease portfolio by tracking terms, deadlines, renewals, billings, tenant leads, vacant-space marketing, rent collection, and leasing activity. Prepare notices and amendments, coordinate with Accounting, Legal, and Property Managers, resolve billing discrepancies, maintain lease records, report portfolio status, and support real estate due diligence.

What you need to know about the San Francisco Tech Scene

San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.

Key Facts About San Francisco Tech

  • Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Google, Apple, Salesforce, Meta
  • Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
  • Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
  • Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account