JPMorgan Chase
Software Engineer III - Machine Learning Platform
$138K–$185K
Palo Alto, CA +3 · Full-time · Posted Aug 20, 2026
- Location
- Palo Alto, CA
Jersey City, NJ
Seattle, WA
New York, NY - Experience
- 3+ years applied experience; no alternatives stated.
- Education
- Formal training or certification on software engineering concepts.
About the job
Builds and operates scalable machine-learning training infrastructure and pipelines, optimizing GPU workloads and enabling governed GenAI/LLM training across cloud environments.
What you'll do
- Design, build, and maintain ML training platforms and Kubernetes infrastructure; run and optimize single-node and distributed GPU workloads.
- Enable GenAI/LLM training and fine-tuning workflows, with evaluation, artifact governance, and enterprise controls.
- Improve observability, automation, developer experience, and operational performance in partnership with data and platform teams.
What they ask for
- Formal training or certification in software engineering concepts and 3+ years of applied experience.
- Experience running ML training in cloud environments and debugging infrastructure and code; hands-on deep-learning workflows with PyTorch or TensorFlow.
- Strong Python skills and experience building automation/CI for ML codebases.
- Experience with distributed training, training performance optimization, Kubernetes, AWS, and enterprise-authorized AI-assisted development tools; understanding of responsible AI use.
Nice to have
- Experience running workloads across multiple cloud platforms; familiarity with cloud-native networking, storage, and artifact management.
- Familiarity with Spark, Ray, Dask, workflow orchestration, model registries, or training cost/performance optimization.
Pay
USD $137,750.00-$185,000.00 base salary (Palo Alto, CA); USD $137,750.00-$185,000.00 base salary (Jersey City, NJ); USD $137,750.00-$185,000.00 base salary (Seattle, WA); USD $137,750.00-$185,000.00 base salary (New York, NY)
Tools
Python, AWS, Kubernetes, EKS, ECR, S3, PyTorch, TensorFlow
AI focus
Builds ML training systems and enables GenAI/LLM training and fine-tuning, including evaluation and governed GPU execution. Uses enterprise-authorized AI coding-assistance tools and validates their outputs.
Model-written summary · the original posting is authoritative