Senior ML Engineer Jobs in Palo Alto, CA

• Build, deploy, and maintain robust pipelines for distributed training on GPU-enabled clusters to support scalable machine learning workflows.
• Develop and manage high-volume real-time and batch inference systems, ensuring optimal performance and reliability.
• Implement quantization techniques and deploy open-weight large language models (LLMs) on modern serving stacks such as vLLM on Ray to maximize efficiency and resource utilization.
• Oversee the management and optimization of vector databases to support advanced AI and machine learning applications.
• Establish and maintain comprehensive monitoring and observability pipelines to ensure system health, performance, and rapid issue resolution.
• Collaborate with cross-functional teams to integrate new technologies and continuously improve existing infrastructure.
• Partner with product, architecture, and other engineering teams to define scalable and performant technical solutions.

• BS in Computer Science or related Engineering field with 10+ years of experience Or MS degree in Computer Science or related Engineering field with 6+ years experience.
• Solid knowledge and extensive experience in Python or in cloud computing and AWS.
• Understanding of quantization techniques such as PTQ, AWQ etc. used to quantize LLMs for accelerating inference on specific GPU architectures
• Solid understanding of Transformer models and challenges involved in serving large transformer-based models
• Solid understanding of ML training, especially latest reinforcement learning algorithms such as GRPO and DAPO
• Experience in systems engineering fundamentals: caching, CUDA, autoscaling, high throughput, low latency, x-region resilient applications
• Deep knowledge and passion for data science fundamentals, training and deploying models
• Experience in monitoring and observability tools to monitor model input/output and features stats
• Solid grounding in engineering fundamentals and analytical mindset

• Experience with recommendation and personalization systems is a plus.
• CUDA experience is a big plus
• Solid fundamentals and experience in containers (docker ecosystem), container orchestration systems [Kubernetes, ECS]
• Experience with Ray, vLLM, RL libraries such as verl/trl
• Good knowledge of Databases

(i) This position is subject to Section 19 of the Federal Deposit Insurance Act. As such, an employment offer for this position is contingent on JPMorganChase's review of criminal conviction history, including pretrial diversions or program entries.

Chase is a leading financial services firm, helping nearly half of America's households and small businesses achieve their financial goals through a broad range of financial products. Our mission is to create engaged, lifelong relationships and put our customers at the heart of everything we do. We also help small businesses, nonprofits and cities grow, delivering solutions to solve all their financial needs.

We offer a competitive total rewards package including base salary determined based on the role, experience, skill set and location. Those in eligible roles may receive commission-based pay and/or discretionary incentive compensation, paid in the form of cash and/or forfeitable equity, awarded in recognition of individual achievements and contributions. We also offer a range of benefits and programs to meet employee needs, based on eligibility. These benefits include comprehensive health care coverage, on-site health and wellness centers, a retirement savings plan, backup childcare, tuition reimbursement, mental health support, financial coaching and more. Additional details about total compensation and benefits will be provided during the hiring process.

We recognize that our people are our strength and the diverse talents they bring to our global workforce are directly linked to our success. We are an equal opportunity employer and place a high value on diversity and inclusion at our company. We do not discriminate on the basis of any protected attribute, including race, religion, color, national origin, gender, sexual orientation, gender identity, gender expression, age, marital or veteran status, pregnancy or disability, or any other basis protected under applicable law. We also make reasonable accommodations for applicants' and employees' religious practices and beliefs, as well as mental health or physical disability needs. Visit our FAQs for more information about requesting an accommodation.

Equal Opportunity Employer/Disability/Veterans