Top 105 Lead machine learning engineer Jobs in Cupertino, Ca

Lead Software Engineer - Machine Learning Platform
JPMorganChase —Palo Alto, CA
- $156,750 - $215,000 a year
- Full-time
- Software troubleshooting
- 5 years
- Senior level
- Health insurance
- Tuition reimbursement
10d
Matterport - Lead Machine Learning R&D Engineer
CoStar Group —Sunnyvale, CA
- $270,000 - $307,000 a year
- Full-time
- 3 years
- Analysis skills
- Senior level
- Employee stock purchase plan
- Health insurance
(General Hire)Tech Lead, Machine Learning Engineer, TikTok Recommendation
TikTok —San Jose, CA
- $254,400 - $480,000 a year
- Computer Science
- 5 years
- Bachelor's degree
- Paid parental leave
- Paid holidays
Lead Machine Learning Engineer-MLOps
JPMorganChase —Palo Alto, CA
- $171,000 - $260,000 a year
- Full-time
- Computer Science
- Bachelor of Science
- Bachelor's degree
- Health insurance
- Tuition reimbursement
Machine Learning Engineer Graduate (Lead Ads) - 2027 Start
TikTok —San Jose, CA
- $128,000 - $256,000 a year
- Multiple hires
- Computer Science
- Office
- Bachelor's degree
- Paid parental leave
- Paid holidays
Staff Machine Learning Engineer, Tech Lead, Labeling Automation
Waymo —Mountain View, CA
- $251,000 - $310,000 a year
- Full-time
- Hybrid work
- 8 years
- C++
- Mentoring
Tech Lead - Machine Learning Platform Engineer
ByteDance —San Jose, CA
- $254,400 - $480,000 a year
- Computer Science
- System design
- Mid-level
- Paid parental leave
- Paid holidays
Lead Machine Learning Engineer
Capital One —San Jose, CA
- Email
- Azure
- Doctoral degree
- Computer Science
- Health insurance
Lead Machine Learning Engineer - Agentic Models, LLM, RAG, GenAI
Eightfold.AI —Santa Clara, CA
- $193,125 - $257,500 a year
- Hybrid work
- Computer science
- Computer Science
- 5 years
- Health insurance
- Dental insurance
Machine Learning Lead - Geospatial, Bellwether
Loon —Mountain View, CA
- $197,000 - $311,000 a year
- Full-time
- Computer science
- 7 years
- Computer Science
Tech Lead, Machine Learning Engineer - TikTok-Data-Search-NLP
TikTok —San Jose, CA
- $254,400 - $480,000 a year
- Computer Science
- 5 years
- Bachelor's degree
- Paid parental leave
- Paid holidays
Lead AI Engineer
Tavant Technologies, Inc. —Santa Clara, CA
- $130,000 - $160,000 a year
- Full-time
- Computer Science
- Research
- Java
- Relocation assistance
- Health insurance
Quick Apply
Tech Lead Machine Learning Engineer - Multimodal/LLM, Trust and Safety
TikTok —San Jose, CA
- $254,400 - $588,000 a year
- Management
- Computer Science
- System design
- Paid parental leave
- Paid holidays
Technical Lead Manager, Machine Learning, GeminiApp Personalization, DeepMind
DeepMind —Mountain View, CA
- $262,000 - $365,000 a year
- Full-time
- Management
- Computer Science
- Engineering
Lead Post-Silicon Validation Engineer
NVIDIA —Santa Clara, CA
- $200,000 - $379,500 a year
- Full-time
- System design
- 11+ years
- Master's degree
Staff Technical Lead Manager, Simulation, Machine Learning
Waymo —Mountain View, CA
- $238,000 - $302,000 a year
- Full-time
- Hybrid work
- Management
- Computer Science
- 5 years
ML Data QA Lead, MLO
Apple —Cupertino, CA
- $120,900 - $249,000 a year
- Analysis skills
- Bachelor's degree
- JavaScript
- Employee stock purchase plan
- Health insurance
21h
Lead RTL Design Engineer
Cerebras Systems —Sunnyvale, CA
- $175,000 - $275,000 a year
- Full-time
- Hybrid work
- Computer Science
- Office
- Master's degree
Principal Lead Distributed Systems Engineer - Services Special Projects
Apple —Cupertino, CA
- $262,500 - $394,000 a year
- Computer science
- Computer Science
- 11+ years
- Employee stock purchase plan
- Health insurance
14d
Senior Product Leader & Strategist, AI Networking Solutions
Hewlett Packard Enterprise —Cupertino, CA
- $194,000 - $412,500 a year
- Full-time
- Computer Science
- Office
- Analysis skills

I want to receive the latest job alert for lead machine learning engineer in cupertino, ca

By signing in to your account, you agree to SimplyHired's Terms of Service and consent to our Cookie and Privacy Policy.

Explore jobs in more locations

lead machine learning engineer jobs near cupertino, ca

Machine Learning Engineer jobs in San Jose, CA

Machine Learning Engineer jobs in Mountain View, CA

System Engineer jobs in Cupertino, CA

Validation Engineer jobs in Santa Clara, CA

Flow Assurance jobs in Houston, TX

Quality Assurance Manager jobs in Hillsborough, NJ

Quality Assurance Manager jobs in Houston, TX

Pharmaceutical Quality Assurance jobs in Remote

Quality Assurance Manager jobs in Sterling, VA

Quality Assurance Manager jobs in Orange, CT

Learning Development jobs in Remote

Lead Machine Learning Engineer jobs in Santa Monica, CA

Lead Machine Learning Engineer jobs in Dallas Fort Worth, TX

Lead Machine Learning Engineer jobs in Troy, MI

Lead Machine Learning Engineer jobs in Saint Petersburg, FL

Machine Learning Engineer New Grad jobs in California

Learning Development jobs in Columbus, OH

Machine Learning Engineer jobs in Columbia, SC

Machine Learning Engineer jobs in Providence, RI

Machine Learning Engineer jobs in Alabama

Lead Software Engineer - Machine Learning Platform

JPMorganChase -
Palo Alto, CA

Apply Now

Job Details

Full-time
$156,750 - $215,000 a year
10 days ago

Benefits

Health insurance
Tuition reimbursement
Retirement plan

Qualifications

AI models
Containerization systems
Performance monitoring
Feature validation testing
Software engineering
Continuous Delivery (CD) implementation
Engineering development testing
Software troubleshooting
Tooling
High-performance computing clusters
Coaching
System performance optimization
AI platforms (beyond public GPTs)
Scalable systems
Computational framework
Machine learning cloud services
Machine intelligence
Model deployment
Release management
Data Security (Data management)
Developing large-scale AI models
Cloud networking
Cloud Native Design
Developing automated testing protocols
Scalability
Model training
Distributed computing
DevOps automation
Machine learning libraries
System tuning

Full Job Description

JOB DESCRIPTION

We have an opportunity to impact your career and provide an adventure where you can push the limits of what's possible.

As a Lead Software Engineer at JPMorganChase within AI/ML Data Platforms, you are an integral part of an agile team that works to build and operate scalable, reliable ML training systems and pipelines on AWS and other cloud platforms. You will productionize training workloads (often GPU-based), improve performance and cost efficiency, and enable repeatable, well-governed training across environments.

Job Responsibilities

Design, build, and maintain end-to-end ML training platform.
Run and optimize GPU training workloads (single-node and distributed), improving throughput, utilization and reproducibility.
Build and operate training infrastructure on Kubernetes (e.g., EKS and other manage Kubernetes platforms), including resource management and workload troubleshooting.
Enable Gen AI/LLM training and fine-tuning workflows (e.g., supervised fine-tuning), including evaluation harnesses, artifact/version governance, and scalable GPU execution patterns aligned to enterprise controls.
Implement observability for training systems: metrics, logs, dashboards, alerting, and operational runbooks.
Partner with data engineering and platform teams to define interfaces, standards, and guardrails (security, access, cost controls)
Improve developer experience for training: standardized containers, CI/CD, templates, documentation, and self-service workflow
Drives team adoption of enterprise-authorized AI-assisted engineering practices within the work environment to improve code quality, delivery speed, and operational outcomes (e.g., AI-assisted code review/refactoring, test strategy acceleration, incident/root-cause analysis support), while establishing consistent validation standards (secure coding, peer review, automated testing) and promoting reuse of effective patterns across the team.
Applies knowledge of tools within the Software Development Life Cycle toolchain, including enterprise-authorized AI-assisted development and automation capabilities, to improve the value realized by automation.

Required qualifications, capabilities, and skills

Formal training or certification on software engineering concepts and 5+ years applied experience.
Demonstrated experience running ML training in cloud environments and debugging issues across infrastructure & code.
Strong Python skills with solid engineering practices (testing, code reviews, modular design, dependency management).
Experience building automation/CI for ML codebases (build, test, release, deployment/promotion workflows).
Hands on experience with deep learning training workflows and at least one major framework (eg., PyTorch or TensorFlow).
Understanding of training performance and stability: data loading bottlenecks, mixed precision, checkpointing, reproducibility, and evaluation methodology.
Experience with distributed training and related concepts (e.g., DDP/FSDP/DeepSpeed concepts, collective communication basics, scaling and bottleneck analysis).
Ability to profile and optimize training systems (CPU/GPU utilization, memory, I/O throughput, networking, scheduling).
Experience with Kubernetes fundamentals for running compute-intensive workloads and AWS (eg., EKS/ECR, S3, IAM, VPC/networking, Cloudwatch, EC2)
Demonstrated experience leading effective use of approved AI-assisted software development tools (e.g., for coding, code review, test acceleration, troubleshooting) with the ability to set team expectations for validating AI outputs for correctness, performance, and security.
Strong understanding of responsible AI use in engineering workflows, including data sensitivity considerations, secure handling of inputs/outputs, and adherence to resiliency and security expectations; experience coaching engineers on safe, compliant adoption within delivery practices.

Preferred qualifications, capabilities, and skills

Experience running training workloads across multiple cloud platforms and managing portability, performance, and governance across environments.
Familiarity with cloud-native networking/storage patterns for high-throughput training and artifact management.
Experience optimizing training input pipelines (sharding, prefetching, caching, format choices such as Parquet/WebDataset) and working with large datasets.
Familiarity with distributed compute frameworks (Spark, Ray, Dask) for feature/dataset generation.
Familiarity with workflow orchestration tools (Airflow-like systems, Argo Workflows-like patterns) and model registry concepts.
Experience optimizing training cost/performance (right-sizing, scheduling policies, interruptible capacity strategies where applicable budge guardrails, quota planning).
Strong observability practice for training systems: metrics/logs/traces, GPU telemetry, dashboards, and alert tuning.

FEDERAL DEPOSIT INSURANCE ACT: 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.

ABOUT US

JPMorganChase, one of the oldest financial institutions, offers innovative financial solutions to millions of consumers, small businesses and many of the world's most prominent corporate, institutional and government clients under the J.P. Morgan and Chase brands. Our history spans over 200 years and today we are a leader in investment banking, consumer and small business banking, commercial banking, financial transaction processing and asset management.

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.

JPMorgan Chase & Co. is an Equal Opportunity Employer, including Disability/Veterans

ABOUT THE TEAM

Our professionals in our Corporate Functions cover a diverse range of areas from finance and risk to human resources and marketing. Our corporate teams are an essential part of our company, ensuring that we're setting our businesses, clients, customers and employees up for success.

Apply Now

Refine Your Search

lead machine learning engineer jobs in cupertino, ca

Lead Software Engineer - Machine Learning Platform

Matterport - Lead Machine Learning R&D Engineer

(General Hire)Tech Lead, Machine Learning Engineer, TikTok Recommendation

Lead Machine Learning Engineer-MLOps

Machine Learning Engineer Graduate (Lead Ads) - 2027 Start

Staff Machine Learning Engineer, Tech Lead, Labeling Automation

Tech Lead - Machine Learning Platform Engineer

Lead Machine Learning Engineer

Lead Machine Learning Engineer - Agentic Models, LLM, RAG, GenAI

Machine Learning Lead - Geospatial, Bellwether

Tech Lead, Machine Learning Engineer - TikTok-Data-Search-NLP

Lead AI Engineer

Tech Lead Machine Learning Engineer - Multimodal/LLM, Trust and Safety

Technical Lead Manager, Machine Learning, GeminiApp Personalization, DeepMind

Lead Post-Silicon Validation Engineer

Staff Technical Lead Manager, Simulation, Machine Learning

ML Data QA Lead, MLO

Lead RTL Design Engineer

Principal Lead Distributed Systems Engineer - Services Special Projects

Senior Product Leader & Strategist, AI Networking Solutions

I want to receive the latest job alert for lead machine learning engineer in cupertino, ca

Related Searches

Explore jobs in more locations

Job Seeker Tools

Employer Tools

Browse

Stay Connected