Senior ML Engineer: LLMs & Scalable Systems

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

Job summary

JPMorgan Chase & Co. is seeking a Senior Machine Learning Engineer in Digital Intelligence to lead research and development of next-generation ML architectures for structured and unstructured financial data.

You will prototype pre-training objectives and implement ideas into robust, scalable production systems. The role emphasizes working with multimodal inputs, advanced transformer models, and large-scale compute, while mentoring engineers and collaborating across teams to deliver real-world

Qualifications

• Requires PhD with 2+ years or Master with 4+ years in Computer Science.
• Deep expertise in LLMs, Transformers, attention mechanisms, RoPE, and multimodal inputs.
• Strong PyTorch skills at scale, distributed training with FSDP/DeepSpeed, memory optimization.

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 and 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

LLMs & Transformers

PyTorch & Distributed Training

Foundation Models training

Software engineering for production

Financial data & recommender systems

Publications in AI/ML venues

Serving infrastructure optimization

Post-training LLMs & network optimize

Large-scale compute infrastructure

Real-world product shipping

Education

PhD in Computer Science

Master's degree in Computer Science

Tools

PyTorch

FSDP

DeepSpeed

Job description

JPMorgan Chase & Co. is seeking a Senior Machine Learning Engineer in Digital Intelligence to lead research and development of next-generation ML architectures for structured and unstructured financial data.

You will prototype pre-training objectives and implement ideas into robust, scalable production systems. The role emphasizes working with multimodal inputs, advanced transformer models, and large-scale compute, while mentoring engineers and collaborating across teams to deliver real-world