> Markdown version of [/jobs/ext/3034958-software-engineer-iii-machine-learning-platform](https://www.wearedevelopers.com/jobs/ext/3034958-software-engineer-iii-machine-learning-platform). Every page supports `.md` or `Accept: text/markdown`. Links point to the HTML versions so they work for humans too. Agent guide: [/agents.md](https://www.wearedevelopers.com/agents.md). --- # Software Engineer III - Machine Learning Platform - **Company:** JPMorgan Chase & Co. - **Location:** Palo Alto, CA, United States - **Experience:** Experienced - **Salary:** $137,750.0 - $185,000.0 - **Contract:** Permanent contract - **Skills:** Microsoft Access, Artificial Intelligence, Airflow, Amazon Web Services, Amazon Elastic Compute Cloud, Amazon S3, Automation of Tests, Unit Testing, Big Data, Cloud Engineering, Code Generation, Software Quality, Code Review, Continuous Integration, Information Engineering, Data Infrastructure, Extract Transform Load (ETL), Software Debugging, Distributed Computing Environment, Identity and Access Management, Python (Programming Language), Machine Learning, Modular Design, Software Tools, Tensorflow, Secure Coding, Software Engineering, Toolchain, Parquet, Cloud Platform System, Pytorch, Large Language Models, Apache Spark, Deep Learning, Caching, Amazon Virtual Private Cloud (VPC), Kubernetes, Dask, Cloudwatch - **Published:** September 23, 2026 - **Apply:** https://www.indeed.com/viewjob?jk=2ce796de85a39f42 ## About the Role * Formal training or certification on software engineering concepts and 3+ 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) * 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. * Understanding of responsible 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 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. ## 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 JPMorganChase within the AI/ML data platform team you serve as a seasoned member of an agile team 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, * 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 * 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., 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. ## Related Videos - [From Messy Queries to Scalable Systems - How Data Engineering actually works](https://www.wearedevelopers.com/videos/100203-from-messy-queries-to-scalable-systems-how-data-engineering-actually-works) - [HTTP headers that make your website go faster](https://www.wearedevelopers.com/videos/1676-http-headers-that-make-your-website-go-faster) - [Introduction to Azure Machine Learning](https://www.wearedevelopers.com/videos/368-introduction-to-azure-machine-learning) - [AI Model Management Life Circles: ML Ops For Generative AI Models From Research to Deployment](https://www.wearedevelopers.com/videos/1152-ai-model-management-life-circles-ml-ops-for-generative-ai-models-from-research-to-deployment) - [Event based cache invalidation in GraphQL](https://www.wearedevelopers.com/videos/433-event-based-cache-invalidation-in-graphql) - [Alibaba Big Data and Machine Learning Technology](https://www.wearedevelopers.com/videos/37-alibaba-big-data-and-machine-learning-technology) ## Related Articles - [How to Become an AI Engineer](https://www.wearedevelopers.com/magazine/331-how-to-become-an-ai-engineer) - [Coffee with Developers - Maria Apazoglou - Making AI understandable for all in production](https://www.wearedevelopers.com/magazine/475-coffee-with-developers-maria-apazoglou-making-ai-understandable-for-all-in-production) - [What Are Large Language Models?](https://www.wearedevelopers.com/magazine/304-what-are-large-language-models) - [Got AI ideas but no money? Here are 10 free ways to level up your AI skills with Google Cloud](https://www.wearedevelopers.com/magazine/600-got-ai-ideas-but-no-money-here-are-10-free-ways-to-level-up-your-ai-skills-with-google-cloud) - [Highest Paying Tech Companies for Developers](https://www.wearedevelopers.com/magazine/220-highest-paying-tech-companies-for-developers) - [MLOps And AI Driven Development](https://www.wearedevelopers.com/magazine/82-mlops-and-ai-driven-development)