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