Principal Software Engineer, LLM Optimization

JPMorgan Chase

Confirmed live today High trust

Quick summary

Work type
On-site
Location
Jersey City, NJNew York, NYSeattle, WAPalo Alto, CA
Employment
Full-time
Posted
61 days ago
Freshness
Confirmed live today

Market check

Salary context

How this pay compares to similar roles

Similar $206k
$127k $291k
below market most similar roles pay here above market

This listing doesn't post a salary. Most similar roles pay $170,250–$241,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 1226 open roles on FindRole.

Listed pay typically runs $156,750–$215,000 across 5 roles with salary data.

Most-posted roles

View all roles at JPMorgan Chase

At a glance

TL;DR · Principal Software Engineer, LLM Optimization

As a Principal Software Engineer - LLM Optimization within the AI/ML Data Platform team, you will serve as the primary technical voice on LLM inference performance. You will own optimization strategies, systematic benchmarking, and efficiency at scale for production workloads. Your daily responsibilities include designing quantization experiments like FP8 and INT8, driving speculative decoding strategies, and building GPU efficiency scorecards. You will evaluate inference engine upgrades, collaborate on KV-cache optimization, and architect agentic AI-enabled engineering workflows. The role requires deep experience with vLLM, TensorRT-LLM, SGLang, and LLM-D, alongside a strong grasp of GPU memory architecture, AWS EKS, and cloud GPU infrastructure. You will solve technical problems regarding throughput, memory reduction, and cost-per-token impact while ensuring production-ready, cost-efficient AI deployment.

What does a Software Engineer earn in New Jersey?

Median $161500 from 34 postings across 14 companies.

See salary data

What you'll do

  • Own systematic benchmarking and performance characterization across all production LLM workloads.
  • Design and execute quantization experiments to measure accuracy, throughput, and cost-per-token impacts.
  • Drive speculative decoding strategies and own acceptance rate measurements across the model portfolio.
  • Build and maintain a GPU efficiency scorecard to monitor utilization, memory headroom, and waste.
  • Lead inference engine upgrade evaluations including new scheduler architectures and tensor parallelism.
  • Collaborate on KV-cache optimization, prefix caching strategies, and multi-node serving architectures.
  • Design and run GPU chaos engineering to test induced failure scenarios and recovery measurements.
  • Architect and govern agentic AI-enabled engineering workflows to improve delivery speed and code quality.

What we're looking for

  • Formal training or certification in software engineering concepts and 7+ years of applied experience.
  • Deep, hands-on experience with LLM inference systems such as vLLM, TensorRT-LLM, SGLang, or LLM-D.
  • Strong grasp of GPU memory architecture, including KV cache sizing, memory-bandwidth vs compute bottlenecks, and quantization implications.
  • Experience with quantization techniques and their real-world tradeoffs at scale.
  • Familiarity with speculative decoding and the variables that drive acceptance rates in production workloads.
  • Rigorous benchmarking instincts using GuideLLM, custom harnesses, or equivalent tools.
  • Comfort operating in cloud GPU infrastructure at scale, specifically AWS, EKS, and managed inference services.
  • Experience with disaggregated prefill/decode serving architectures, GPU hardware diagnostics, and ML observability (preferred).

More like this

Similar roles

Principal Software Engineer

JPMorgan Chase

Jersey City, NJ +1 99 days ago
Python Java LLM RAG Agentic AI AWS Azure GCP Kafka Redis MemCached Airflow Temporal Spark PySpark Databricks Snowflake Apache Iceberg Dynatrace Splunk Grafana Microservices API Design Vector Stores
7+ yrs exp

Principal Software Engineer

Autodesk

Atlanta, GA 74 days ago
Distributed Systems Data Pipelines NoSQL Object Storage Serverless API Design ETL DevOps Search Relevance Schema Validation
8+ yrs exp Hybrid

Principal Software Engineer

Amgen

Tampa, FL +1 139 days ago
Generative AI LLMs MLOps LLMOps Python PyTorch TensorFlow Hugging Face LangChain AWS Azure GCP Kubernetes Docker Apache Spark Apache Airflow MLflow Kubeflow Databricks RAG Vector Databases CI/CD Distributed Systems NLP
8+ yrs exp Hybrid

Principal Software Engineer

The Walt Disney Company

San Francisco, CA +1 16 days ago $209,500–$280,900
Video Coding C++ Python AV1 AV2 AVC HEVC VVC Computer Vision Deep Learning gStreamer Docker REST APIs XML JSON Git GitHub ABR Algorithms
10+ yrs exp Hybrid

Principal Software Engineer

The Walt Disney Company

San Francisco, CA +1 16 days ago $209,500–$280,900
Video Coding AV1 AV2 AVC HEVC VVC Deep Learning Computer Vision C++ Python gStreamer Docker REST APIs XML JSON Git GitHub ABR Algorithms Video Processing
10+ yrs exp Hybrid

Principal Software Engineer

State Street

Burlington, MA 106 days ago
Java C# React SQL JavaScript Azure Jenkins Git REST SOAP XML JSON CI/CD Agile Scrum PowerShell Batch RDBMS Micro services BDD ATDD
10+ yrs exp Hybrid