Lead Software Engineer, Machine Learning
JPMorgan Chase
Quick summary
Market check
How this pay compares to similar roles
This listing doesn't post a salary. Most similar roles pay $192,050–$238,250.
Based on 240 similar postings.
Employer
JPMorgan Chase & Co. is a global financial services firm and one of the largest banks in the world, offering investment banking, commercial banking, asset management, and consumer financial services.
JPMorgan Chase currently has 1117 open roles on FindRole.
Listed pay typically runs $186,160–$215,000 across 7 roles with salary data.
Most-posted roles
At a glance
Lead Software Engineer - Machine Learning Platform joins the AI/ML Data Platforms team to build and operate scalable, reliable ML training systems and pipelines on AWS and other cloud platforms. This role involves productionizing GPU-based workloads, optimizing performance and cost efficiency, and enabling well-governed training across environments. Key responsibilities include designing end-to-end training platforms, managing Kubernetes infrastructure like EKS, and supporting Gen AI/LLM fine-tuning workflows with robust evaluation harnesses. The candidate will implement observability for training systems, improve developer experience through CI/CD and standardized containers, and lead the adoption of AI-assisted engineering practices. Required skills include Python, PyTorch or TensorFlow, distributed training concepts like DDP and DeepSpeed, and expertise in Kubernetes and AWS services. The role addresses technical challenges in GPU utilization, data loading bottlenecks, and large-scale dataset management for enterprise-grade machine learning infrastructure.
Skills
What you'll do
What we're looking for
More like this
JPMorgan Chase
JPMorgan Chase
Capital One Financial
JPMorgan Chase
Stripe
Stripe