JPMorgan Chase

Applied AI ML Lead - AI Agents & Agentic Systems

JPMorgan Chase
Apply
11 days ago
Palo Alto, CA, USAStaff+
H1B sponsor

Base Salary

$171k - $260k/yr

Responsibilities

  • Serve as a subject matter expert on ML algorithms, frameworks, techniques, and optimizations.
  • Conduct experiments and benchmarks involving AI, LLM, and agentic techniques, including model and system tuning.
  • Develop Python code for proof-of-concept experiments and production-ready solutions.
  • Productionize and integrate machine learning and generative AI solutions within the ML platform.
  • Optimize accuracy and performance by identifying and resolving system inefficiencies and bottlenecks.
  • Collaborate with product, engineering, data, and ML teams to deliver business-focused solutions.
  • Drive decisions affecting product design, application functionality, and technical operations.
  • Mentor engineers and ML practitioners and raise technical standards through strong engineering practices.

Requirements

  • MS or PhD in Computer Science, Machine Learning, or a related field.
  • At least 5 years of applied machine learning experience.
  • At least 5 years of experience with a programming language such as Python, Java, or C/C++; intermediate Python proficiency is required.
  • At least 5 years of experience applying data science and ML techniques to business problems.
  • Strong knowledge of LLMs and agentic applied science, including multi-agent orchestration, reasoning, skills, and tools.
  • Experience with applied research and experimentation in machine learning and deep learning, including LLMs and generative AI.
  • Deep expertise in PyTorch or TensorFlow.
  • Experience with GPU optimization, fine-tuning, embedding models, inference, prompt engineering, evaluation, RAG, reasoning, context management, and agentic capabilities.
  • Preferred experience with distributed training frameworks, search and ranking, recommender systems, graph techniques, multi-agent orchestration, evaluation, and benchmarking.
  • Preferred advanced knowledge of reinforcement learning or meta-learning.
  • Preferred experience deploying ML models on AWS and tools such as SageMaker and EKS.
  • Ability to complete projects with limited supervision and demonstrate leadership when working with engineers, product managers, and ML practitioners.

Benefits

  • Competitive total rewards package with role-, experience-, skill-, and location-based base salary.
  • Eligible roles may receive commission-based pay and/or discretionary incentive compensation in cash or forfeitable equity.
  • Benefits may include comprehensive healthcare, on-site health and wellness centers, retirement savings, backup childcare, tuition reimbursement, mental health support, and financial coaching.
  • Additional compensation and benefits details are provided during the hiring process.
JPMorgan Chase

About JPMorgan Chase

10,000+ employees

JPMorgan Chase provides consumer and commercial banking, payments, credit card, wealth management, and corporate and investment banking services to individuals, businesses, institutions, and governments. The public company (NYSE: JPM) earns revenue from interest, fees, trading, and asset management across operations in more than 100 markets. Headquartered in New York City with roots dating to 1799, it serves retail customers and prominent corporate and government clients through brands including Chase and J.P. Morgan.

Contact me