Lead Machine Learning Engineer

JPMorgan Chase & Co.; New York (NY); On-site; USD 150,000 - 210,000; Full time

Job summary

JPMorgan Chase & Co. in New York seeks a Lead Machine Learning Engineer to design, build, and scale ML pipelines for production models across digital channels.

You will work with a high-caliber team of software developers and ML experts, focusing on distributed training on GPU clusters, model monitoring, and robust deployment.

The role requires strong Python, cloud, and ML framework experience, plus hands-on tooling such as Ray, Spark, Docker, Kubernetes, and Airflow.

Qualifications

• BS in Computer Science or related Engineering with 6+ years of experience or MS with 4+ years.
• Strong Python and cloud computing experience, plus ML frameworks (PyTorch, TensorFlow).
• Deep knowledge of data science fundamentals and model deployment.
• Experience with monitoring/observability tools for models.
• Experience with Ray, Spark, and training/inference systems.
• Solid grounding in engineering fundamentals and enterprise design.

Responsibilities

• Build, deploy, and maintain distributed training pipelines on GPU clusters.
• Develop pipelines for model promotion and MDLC-related capabilities.
• Optimize training throughput for large data sources.
• Integrate platforms and tools for model monitoring and observability.
• Collaborate with cross-functional teams to advance ML Platform capabilities.
• Partner with product and engineering to power digital channels.

Skills

Python

Cloud computing

ML frameworks

Model training

Monitoring observability

Big data tools

System design

Education

BS in Computer Science or related Engineering

MS in Computer Science or related Engineering

Tools

Docker

Kubernetes

Airflow

Kubeflow

Ray

Job description

Consumer & Community Banking division serves our Chase customers through a range of financial services, including personal banking, credit cards, mortgages, auto financing, investment advice, small business loans and payment processing. We're proud to lead the U.S. in credit card sales and deposit growth and have the most-used digital solutions - all while ranking first in customer satisfaction. In this role, you'll apply strong technical judgment to choose the right approaches (including modern LLM-based methods where appropriate), evaluate performance with rigorous metrics, and ensure solutions are reliable, secure, and scalable in real-world environments. You'll also contribute to improving data quality and feedback loops, monitoring models in production, and continuously iterating to reduce agent effort, shorten resolution times, and increase consistency and quality across operational workflows.

As Lead Machine Learning Engineer on the Digital Intelligence team, you will be collaborating with a high-caliber team of software developers and deep learning experts, you'll build and maintain pipelines for distributed model training on large compute clusters, hyperparameter tuning at scale, model monitoring, design and develop ML frameworks and components used for various model implementations.

Job responsibilities

• Build, deploy, and maintain robust pipelines for distributed training on GPU-enabled clusters to support scalable machine learning workflows.
• Develop and manage pipelines for model promotion and other capabilities related to MDLC.
• Optimize training throughput for large data sources
• Establish and maintain integrations to platforms and tools related to model monitoring and observability
• Collaborate with cross-functional teams to integrate new technologies and improve the capabilities of our ML Platform.
• Partner with product, architecture, modeling, and engineering to design robust solutions that power our Digital channels

Required qualifications, capabilities, and skills

• BS in Computer Science or related Engineering field with 6+ years of experience Or MS degree in Computer Science or related Engineering field with 4+ years experience.
• Solid knowledge and extensive experience in Python and in cloud computing, along with ML frameworks (i.e. pytorch, tensorflow)
• Deep knowledge and passion for data science fundamentals, training and deploying models
• Experience in monitoring and observability tools to monitor model input/output and features stats
• Operational experience in big data/ML tools such as Ray, Spark andin training/inference systems such asRay, vllm/SGLang
• Solid grounding in engineering fundamentals and enterprise system design

Preferred qualifications, capabilities, and skills

• Experience with recommendation and personalization systems is a plus.
• CUDA experience is a big plus
• Solid fundamentals and experience in containers (docker ecosystem), container orchestration systems [Kubernetes, ECS], DAG orchestration [Airflow, Kubeflow etc]
• Good knowledge of data storage solutions and strategies (online and offline)