[Skip to main content](#main-content) [OpenNash Jobs](https://jobs.opennash.com/) [Hiring report](/reports/hiring/) · [All JPMorganChase jobs](https://jobs.opennash.com/companies/jpmorganchase/) · [Search jobs](/search/) JPMorganChase · Palo Alto, CA # Lead Software Engineer - Machine Learning Platform Palo Alto, CA, United States Manager or lead Software Engineering Listed on JPMorganChase’s official careers site. Posted August 10, 2026. Check that it is still open and apply there. [Apply on JPMorganChase’s careers site →](https://jpmc.fa.oraclecloud.com/hcmUI/CandidateExperience/en/sites/CX_1001/job/210772141) [OpenNash Jobs](https://jobs.opennash.com/) / [JPMorganChase](https://jobs.opennash.com/companies/jpmorganchase/) / Lead Software Engineer - Machine Learning Platform ## Job description We have an opportunity to impact your career and provide an adventure where you can push the limits of what's possible. As a Lead Software Engineer at JPMorganChase within AI/ML Data Platforms, you are an integral part of an agile team that works to build and operate scalable, reliable ML training systems and pipelines on AWS and other cloud platforms. You will productionize training workloads (often GPU-based), improve performance and cost efficiency, and enable repeatable, well-governed training across environments. ### Job Responsibilities Design, build, and maintain end-to-end ML training platform. Run and optimize GPU training workloads (single-node and distributed), improving throughput, utilization and reproducibility. Build and operate training infrastructure on Kubernetes (e.g., EKS and other manage Kubernetes platforms), including resource management and workload troubleshooting. Enable Gen AI/LLM training and fine-tuning workflows (e.g., supervised fine-tuning), including evaluation harnesses, artifact/version governance, and scalable GPU execution patterns aligned to enterprise controls. ### Implement observability for training systems metrics, logs, dashboards, alerting, and operational runbooks. Partner with data engineering and platform teams to define interfaces, standards, and guardrails (security, access, cost controls) ### Improve developer experience for training standardized containers, CI/CD, templates, documentation, and self-service workflow Drives team adoption of enterprise-authorized AI-assisted engineering practices within the work environment to improve code quality, delivery speed, and operational outcomes (e.g., AI-assisted code review/refactoring, test strategy acceleration, incident/root-cause analysis support), while establishing consistent validation standards (secure coding, peer review, automated testing) and promoting reuse of effective patterns across the team. 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. Required qualifications, capabilities, and skills Formal training or certification on software engineering concepts and 5+ years applied experience. Demonstrated experience running ML training in cloud environments and debugging issues across infrastructure & code. Strong Python skills with solid engineering practices (testing, code reviews, modular design, dependency management). Experience building automation/CI for ML codebases (build, test, release, deployment/promotion workflows). Hands on experience with deep learning training workflows and at least one major framework (eg., PyTorch or TensorFlow). ### Understanding of training performance and stability data loading bottlenecks, mixed precision, checkpointing, reproducibility, and evaluation methodology. Experience with distributed training and related concepts (e.g., DDP/FSDP/DeepSpeed concepts, collective communication basics, scaling and bottleneck analysis). Ability to profile and optimize training systems (CPU/GPU utilization, memory, I/O throughput, networking, scheduling). Experience with Kubernetes fundamentals for running compute-intensive workloads and AWS (eg., EKS/ECR, S3, IAM, VPC/networking, Cloudwatch, EC2) Demonstrated experience leading effective use of approved AI-assisted software development tools (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 engineers on safe, compliant adoption within delivery practices. Preferred qualifications, capabilities, and skills Experience running training workloads across multiple cloud platforms and managing portability, performance, and governance across environments. Familiarity with cloud-native networking/storage patterns for high-throughput training and artifact management. Experience optimizing training input pipelines (sharding, prefetching, caching, format choices such as Parquet/WebDataset) and working with large datasets. Familiarity with distributed compute frameworks (Spark, Ray, Dask) for feature/dataset generation. Familiarity with workflow orchestration tools (Airflow-like systems, Argo Workflows-like patterns) and model registry concepts. Experience optimizing training cost/performance (right-sizing, scheduling policies, interruptible capacity strategies where applicable budge guardrails, quota planning). ### Strong observability practice for training systems metrics/logs/traces, GPU telemetry, dashboards, and alert tuning. ### 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. ## At a glance - Posted by the employer: Aug 10, 2026 - First seen by OpenNash: Aug 21, 2026 - Last checked: Oct 13, 2026 - Listed for: 64 days - Location: Palo Alto, CA, United States[Apply on JPMorganChase’s careers site →](https://jpmc.fa.oraclecloud.com/hcmUI/CandidateExperience/en/sites/CX_1001/job/210772141) ## More jobs at JPMorganChase ### [Lead Software Engineer - API Platform](https://jobs.opennash.com/companies/jpmorganchase/jobs/7bdef74a94a23657b30e/) Palo Alto, CA, United States ### [Lead Software Engineer - Full Stack/Java/ React](https://jobs.opennash.com/companies/jpmorganchase/jobs/0a01b24af0587fb26fdb/) Palo Alto, CA, United States ### [Lead Software Engineer - Java, AWS](https://jobs.opennash.com/companies/jpmorganchase/jobs/332610871c2bff9cb169/) Palo Alto, CA, United States ### [Lead Software Engineer - Machine Learning](https://jobs.opennash.com/companies/jpmorganchase/jobs/815edc9fa3516a660061/) Palo Alto, CA, United States ### [Lead Software Engineer -Infrastructure Technical Duty Officer](https://jobs.opennash.com/companies/jpmorganchase/jobs/5c4eebf6e5df65576ffb/) Palo Alto, CA, United States ## Similar jobs at other companies ### [Software Development Engineer (Machine Learning)](https://jobs.opennash.com/companies/fortinet/jobs/b767b5cf9b10bdd48539/) Fortinet · Sunnyvale, CA, United States ### [Staff Software Engineer, On-Device Machine Learning](https://jobs.opennash.com/companies/alphabet/jobs/a163c36c96d6d5b846cf/) Alphabet · Sunnyvale, CA, USA ### [Senior Machine Learning Software Engineer](https://jobs.opennash.com/companies/qualcomm/jobs/1bc927a217a6f7d43f7e/) Qualcomm · Boulder, Colorado, United States of America ### [Senior Machine Learning Software Engineer](https://jobs.opennash.com/companies/qualcomm/jobs/0ef4466cd07c6ef265b5/) Qualcomm · San Diego, California, United States of America ### [Spring 2027 - Software Development & Machine Learning - Collegiate Intern](https://jobs.opennash.com/companies/general-dynamics/jobs/91fd79d8266c68eac683/) General Dynamics · Savannah, GA, US OpenNash lists jobs from employers’ own careers sites and does not handle applications. Dates come from the employer where stated; “first seen” and “last checked” are our own observations. OpenNash Jobs · Official listings with dates and links to the employer. [Weekly hiring report](/reports/hiring/) · [Methodology](/methodology/)