ML Engineer - LLMs for Production & Scale

JPMorgan Chase & Co.; Palo Alto (CA); On-site; USD 180,000 - 240,000; Full time

Job summary

JPMorgan Chase & Co. is seeking a Machine Learning Engineer-Digital Intelligence in the Consumer & Community Banking division.

You will apply advanced ML and LLM techniques to build reliable, scalable, production-grade models that interact with financial data and product use cases. You will collaborate with software engineers and deep learning experts to optimize training, evaluate performance with robust metrics, and translate research into deployable systems in a fast-paced, collaborative

Qualifications

• Master’s degree with 2+ years or Bachelor’s with 4+ years in Computer Science, with training and work experience in Machine Learning, LLM/NLP or similar fields.
• Deep LLM and Transformer expertise — attention mechanisms, RoPE, multi-modal data
• PyTorch proficiency at scale with distributed training frameworks (FSDP/DeepSpeed) and memory optimization
• Foundation model training — pre-training from scratch and token/vocabulary design for heterogeneous data sources (tabular, temporal, graphical)
• Strong software engineering skills — production-quality systems that run reliably at scale
• Experience with financial data and recommendation systems

Responsibilities

• Research and prototype next-generation architectures for structured and unstructured data
• Develop novel pre-training objectives for financial event sequences and heterogeneous profile data
• Implement research ideas in production-quality code
• Mentor engineers on ML best practices; translate research advances into deployable systems
• Optimize training throughput for large data sources
• Collaborate with other teams to design solutions for product use cases

Skills

LLM/NLP

PyTorch

Distributed training

Memory optimization

Production systems

Financial data experience

Recommendation systems

Software engineering

Education

Master’s degree in Computer Science

Bachelor’s degree in Computer Science

Tools

PyTorch

FSDP

DeepSpeed

Job description

JPMorgan Chase & Co. is seeking a Machine Learning Engineer-Digital Intelligence in the Consumer & Community Banking division.

You will apply advanced ML and LLM techniques to build reliable, scalable, production-grade models that interact with financial data and product use cases. You will collaborate with software engineers and deep learning experts to optimize training, evaluate performance with robust metrics, and translate research into deployable systems in a fast-paced, collaborative