Drimva

JPMorgan Chase

Senior Machine Learning Engineer – Digital Intelligence

$171K–$260K

Palo Alto, CA · Senior · Full-time · Posted Aug 6, 2026

Experience
PhD with 2+ years OR Master’s degree with 4+ years.
Education
PhD or Master’s degree in Computer Science.

About the job

Researches and builds large language and machine learning systems for financial data, turning new methods into scalable production products and improving operational workflows.

What you'll do

  • Research and prototype architectures and pre-training objectives for structured, unstructured, and financial event data.
  • Implement research in production-quality code and optimize training throughput for large data sources.
  • Mentor engineers and collaborate across teams to design deployable product solutions.

What they ask for

  • PhD with 2+ years or Master’s degree with 4+ years in Computer Science, with training and work experience in Machine Learning, LLM/NLP or similar fields.
  • Deep LLM and Transformer expertise, including attention mechanisms, positional encodings such as RoPE, and multimodal inputs.
  • PyTorch proficiency at scale, including distributed training with FSDP and DeepSpeed and practical memory optimization.
  • Foundation-model pre-training from scratch, token and vocabulary design for heterogeneous data, strong software engineering skills, and prior experience with financial data and recommendation systems.

Nice to have

  • Publication record at top AI/ML venues.
  • Experience optimizing serving infrastructure, post-training LLMs, network optimization, interpretability or steering, large-scale compute, or shipping real-world products.

Pay

$171,000.00-$260,000.00 USD base salary

Tools

LLM, Transformer, RoPE, PyTorch, FSDP, DeepSpeed

AI focus

Develops and optimizes LLMs and other machine learning models, including pre-training, multimodal and heterogeneous-data modeling, post-training, interpretability, and production deployment.

Model-written summary · the original posting is authoritative

First seen Oct 28, 2026 · Last verified Nov 5, 2026 (UTC)