J

JP Morgan Chase

Runs a consumer bank and an investment bank

Machine Learning Engineer - Digital Intelligence

Natural Language Processing & LLMsConfirmed live in the last 24 hours
Full-timePalo Alto, CA, USA$137.8k - $195kIn-person

About the Job

Requirements

  • A Master’s degree with at least 2 years of experience or a Bachelor’s degree with at least 4 years of experience in Computer Science, with training and work experience in Machine Learning, large language models/natural language processing, or similar fields.
  • Deep expertise in large language models and Transformers, including attention mechanisms, positional encodings such as RoPE, and handling multimodal data inputs.
  • Proficiency in PyTorch at scale and hands-on experience with distributed training frameworks including FSDP and DeepSpeed, alongside practical memory optimization techniques.
  • Proven experience pre-training foundation models from scratch and designing tokens and vocabularies for complex, heterogeneous data sources, including tabular, temporal, and graphical formats.
  • Strong software engineering skills and the ability to build robust, production-quality systems that perform reliably at scale.
  • Prior experience with financial data and recommendation systems.

Responsibilities

  • Research and prototype next-generation architectures for structured and unstructured data.
  • Develop novel pre-training objectives tailored to financial event sequences and heterogeneous profile data.
  • Implement research ideas in production-quality code.
  • Mentor engineers on machine learning 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.

Desired Qualifications

  • A publication record at top AI/ML venues.
  • Experience optimizing serving infrastructure.
  • Experience with post-training large language models and network optimization algorithms, as well as interpretability or steering techniques for large language models.
  • Experience working with large-scale compute infrastructure.
  • Experience shipping a real-world product, project, or feature.
  • Experimental rigor and ablation design when benchmarking large language model optimizations.

About the Company

JP Morgan Chase

Runs a consumer bank and an investment bank

JPMorgan Chase runs the largest bank in the United States, which serves households and small businesses as Chase and large companies, governments, and investors as J.P. Morgan. During the 2008 financial crisis it bought the failing investment bank Bear Stearns and lender Washington Mutual, and in 2023 it took over First Republic Bank after regulators seized it. Chase takes deposits and earns interest on credit cards, mortgages, and car loans, while the investment side earns fees for advising on mergers, raising money for companies, trading stocks and bonds, and managing wealth. Formed in a 2000 merger, JPMorgan Chase traces its roots to 1799 and is based in New York City.

Company size
10,001+
Company stage
IPO
Headquarters
New York City, New York
Founded
1799

Recently Posted Jobs

Explore more opportunities at JP Morgan Chase

See more jobs