Lead Machine Learning Engineer, MLOps

JPMorgan Chase

Confirmed live yesterday High trust

Quick summary

Work type
On-site
Location
Palo Alto, CA
Posted
72 days ago
Freshness
Confirmed live yesterday

Market check

Salary context

How this pay compares to similar roles

Similar $225k
$174k most similar roles pay here $285k

This listing doesn't post a salary. Most similar roles pay $195,787–$254,750.

Based on 240 similar postings.

Employer

About JPMorgan Chase

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

View all roles at JPMorgan Chase

At a glance

TL;DR · Lead Machine Learning Engineer, MLOps

Lead Machine Learning Engineer-MLOps will join the Recommendation Engine team to build and deploy machine learning models on a modern MLOps stack. This role involves building and maintaining pipelines for distributed model training on GPU-enabled clusters, managing batch and real-time inference, performing hyperparameter tuning at scale, and implementing quantization techniques for large language models. The position also requires overseeing vector database optimization and establishing monitoring and observability pipelines to ensure system health. Key technologies include Python, AWS, Ray, DuckDB, Spark, vllm/SGLang, and the Docker ecosystem with Kubernetes or ECS. Candidates will utilize Airflow or Kubeflow for DAG orchestration while addressing technical challenges in high-throughput, low-latency applications. The work focuses on the Personalization and Insights product area, specifically building systems that power personalized experiences across various banking and merchant offer channels.

What you'll do

  • Build and maintain pipelines for distributed model training on GPU-enabled clusters.
  • Develop high-throughput batch and real-time inference pipelines for production deployment.
  • Implement quantization techniques to optimize the performance and efficiency of large language models.
  • Manage and optimize vector databases to support advanced AI applications.
  • Establish monitoring and observability pipelines to track system health and model performance.
  • Integrate new technologies into existing infrastructure to improve overall system reliability.
  • Design scalable technical solutions for high-throughput, low-latency personalization systems on AWS.

What we're looking for

  • A BS in Computer Science or related field with 6+ years of experience.
  • An MS degree in Computer Science or related field with 4+ years of experience.
  • Extensive experience in Python and cloud computing, preferably AWS.
  • Experience with systems engineering fundamentals including caching, CUDA, autoscaling, high throughput, and low latency.
  • Knowledge of quantization techniques like PTQ and AWQ for accelerating LLM inference.
  • Proficiency with big data/ML tools such as Ray, DuckDB, Spark, vllm, and SGLang.
  • Experience in monitoring and observability tools to track model performance and feature statistics.
  • Familiarity with container orchestration (Kubernetes, ECS) and DAG orchestration (Airflow, Kubeflow).

More like this

Similar roles

Lead Machine Learning Engineer

Capital One Financial

New York, NY 37 days ago $215,200$245,600
Machine Learning Python Scala Java PyTorch TensorFlow scikit-learn Spark Dask AWS Azure Google Cloud Platform Data Pipelines Distributed Computing Agile
6+ yrs exp

Lead Machine Learning Engineer

Capital One Financial

Cambridge, MA +2 43 days ago $197,300$225,100
Python Scala Java PyTorch TensorFlow scikit-learn Spark Dask AWS Bedrock Google Cloud Azure RAG Data Pipelines Distributed Computing Agile
6+ yrs exp

Lead Machine Learning Engineer

Capital One Financial

Plano, TX 66 days ago $179,400$204,700
Generative AI LLM RAG LangChain LangGraph LlamaIndex Python Scala Java PyTorch TensorFlow scikit-learn Spark Dask Vector Databases AWS Azure Google Cloud Platform Distributed Computing
6+ yrs exp

Lead Machine Learning Engineer

Capital One Financial

San Jose, CA +4 65 days ago $197,300$225,100
Machine Learning Python Go Scala Java AWS Kubernetes PyTorch TensorFlow scikit-learn Spark Dask Data Pipelines Agile Distributed Computing
6+ yrs exp

Lead Machine Learning Engineer

The Walt Disney Company

Remote (CA) 60 days ago $187,900$252,000
Machine Learning LLM RecSys PyTorch TensorFlow Databricks Spark SQL Reinforcement Learning Algorithm Development
7+ yrs exp Remote