ZipRecruiter

Log In

1

Staff Machine Learning Engineer Jobs in California

Walmart

Staff, Machine Learning Engineer

Sunnyvale, CA · On-site

$130K - $260K/yr

We are looking for a strong Staff Machine Learning Engineer who has the passion to develop AI driven intelligent products for the Associate Productivity and Experience team, with the ability to ...

Shipt

Staff Machine Learning Engineer

San Francisco, CA · On-site +1

Impact As a Staff Machine Learning Engineer on Shipt's Personalization Platform team you will drive key AI initiatives. In this role, you'll collaborate with Data Scientists to design and deploy ...

New

Apple

Staff Machine Learning Engineer

Cupertino, CA · On-site

... engineers and researchers the data systems and large-scale compute they need to build and ship models at Apple's bar for quality and privacy. Strong foundation in machine learning, with hands-on ...

Uber

Staff Machine Learning Engineer

San Francisco, CA · On-site

... machine learning development productivity across Uber. - Work closely with engineers in the broader Uber ML/AI Platform Team (Michelangelo) to improve the broader ML Platform ecosystem for our users ...

Uber

Staff Machine Learning Engineer

San Francisco, CA · On-site +1

... machine learning development productivity across Uber.- Work closely with engineers in the broader Uber ML/AI Platform Team (Michelangelo) to improve the broader ML Platform ecosystem for our users ...

Geico

Machine Learning Engineer II

Palo Alto, CA

Staff Machine Learning Engineer Overview: As a Staff Machine Learning Engineer, you will be the overall tech lead of a single AI/Machine Learning team, responsible for the tech design and tech health ...

New

Nubank

Staff Machine Learning Engineer

Palo Alto, CA · On-site

$230K - $345K/yr

We are seeking Machine Learning Engineers to lead high-impact research projects that bridge the gap between state-of-the-art AI and production-grade financial systems. You will be responsible for ...

Uber

Staff Machine Learning Engineer

San Francisco, CA · On-site +1

... programming languages such as Python, Spark SQL, Presto, Go, Java**Why Join Us?**- Work on high-impact machine learning problems that directly improve Uber's marketplace efficiency.- Influence key ...

next page

Showing results 1-20

People also search for

All JobsStaff Machine Learning Engineer JobsStaff Machine Learning Engineer Jobs in California

Staff Machine Learning Engineer information

See California salary details

$22.7K

$98K

$190K

How much do staff machine learning engineer jobs pay per year?

As of May 29, 2026, the average yearly pay for staff machine learning engineer in California is $98,029.00, according to ZipRecruiter salary data. Most workers in this role earn between $68,100.00 and $123,400.00 per year, depending on experience, location, and employer.

What is a Staff Machine Learning Engineer job?

A Staff Machine Learning Engineer is a senior-level technical role responsible for designing, deploying, and optimizing machine learning models at scale. They provide technical leadership, mentor other engineers, and drive best practices in ML system architecture. This role often involves collaborating with cross-functional teams, improving model performance, and ensuring the reliability of machine learning solutions in production. Staff ML Engineers typically have deep expertise in algorithms, data infrastructure, and engineering processes. Their work focuses on solving complex problems and influencing the broader ML strategy within an organization.

What are the key skills and qualifications needed to thrive in the Staff Machine Learning Engineer position, and why are they important?

To thrive as a Staff Machine Learning Engineer, you need deep expertise in machine learning algorithms, software engineering, data analysis, and typically a strong academic background in computer science or related fields. Experience with Python, TensorFlow, PyTorch, cloud platforms, and a track record of delivering production-level ML systems are crucial, as are advanced degrees or relevant certifications. Strong leadership, communication, and mentoring skills help you effectively guide teams and collaborate across departments. These competencies are essential for designing robust ML solutions, leading technical initiatives, and ensuring successful project delivery in complex organizational environments.

What are the typical collaboration and leadership responsibilities for a Staff Machine Learning Engineer?

As a Staff Machine Learning Engineer, you often serve as a technical leader, partnering with cross-functional teams including data scientists, product managers, and software engineers to develop and deploy machine learning solutions. You will mentor junior engineers, conduct code reviews, and help establish best practices for model development and deployment. In addition to hands-on technical work, you may be responsible for evaluating new tools, contributing to the broader ML strategy, and facilitating knowledge sharing sessions. This collaborative and leadership-focused approach helps ensure consistency, quality, and innovation across machine learning projects.
More about Staff Machine Learning Engineer jobs
What are the most commonly searched types of Staff Machine Learning Engineer jobs in California? The most popular types of Staff Machine Learning Engineer jobs in California are:
What are popular job titles related to Staff Machine Learning Engineer jobs in California? For Staff Machine Learning Engineer jobs in California, the most frequently searched job titles are:
What job categories do people searching Staff Machine Learning Engineer jobs in California look for? The top searched job categories for Staff Machine Learning Engineer jobs in California are:
What cities in California are hiring for Staff Machine Learning Engineer jobs? Cities in California with the most Staff Machine Learning Engineer job openings:
Staff Machine Learning Engineer jobs near you
Staff, Machine Learning Engineer

Staff, Machine Learning Engineer

Walmart

Sunnyvale, CA • On-site

$130K - $260K/yr

Other

Medical, Dental, Vision, Life, Retirement, PTO

Posted 20 days ago

Walmart rating

6.0

Company rating: 6.0 out of 10

Based on 21,562 frontline employees who took The Breakroom Quiz

22nd of 39 rated national retailers

Job description

Position Summary...What you'll do...What you'll do: We are looking for a strong Staff Machine Learning Engineer who has the passion to develop AI driven intelligent products for the Associate Productivity and Experience team, with the ability to influence and collaborate with partner teams, data scientists, ML engineers, product managers and other engineers in the team. A successful candidate will have an established hands-on data science, AI, ML experience in driving technical products, Full stack developer, and a passion for great customer-centric products.
  • Design, architect, build AI/ML models, AI Agents and deploy, operate, optimize the solutions
  • Work on Python and mainstream machine learning frameworks, e.g. TensorFlow or PyTorch and Agentic Frameworks like Pydantic
  • Lead development of end-to-end system architecture for GenAI/AI/ML and data intensive applications
  • Design, Develop and deploy Full stack based applications.
  • Develop and deploy production-grade real-time/batch machine learning services Work with product managers to design user journeys, feedback loop and analyze user telemetry
  • Identify or propose AI/ML use-cases to business teams to improve business processes and develop quick MVPs/ POCs to help stakeholders make better decisions
  • End to end ownership of product using technologies such as Java , Node JS, React Js
  • You will be required to review design and code, provide technical solutions to team for complex requirements
  • Uphold and enforce rigorous standards, governance, and best practices, ensuring strict compliance with model and data governance standards.
  • Adhere strictly to Walmart's policies, procedures, mission, values, standards of ethics, and integrity.
  • Innovation and Research: Stay at the forefront of technological advancements, conducting research to identify emerging tools and methodologies.
  • Collaborative Communication: Communicate complex technical concepts effectively to both technical and non-technical stakeholders, promoting a culture of open communication and collaboration.
  • You will be called upon to support the coaching and training of other team members to ensure all employees are confident in the use of software applications
What you'll bring:
  • bachelor's/master's degree in computer science, information technology, engineering, or related area
  • Minimum 10 years of experience in design, development, deployment, and maintenance of highly scalable, robust, and user-centric Fullstack/GenAI /AI /ML products and applications
  • Good understanding of Data Structures
  • Hands on experience in systems design, algorithms, and distributed systems.
  • Industry level experience in developing production-grade AI/ML system development with major technology companies or AI-native start-ups.
  • Experience at writing high-performing production-quality code in Python
  • Exposure to cloud infrastructure, such as Open Stack, Azure, GCP, or AWS
  • Experience working in Agile Methodology
  • Experience using CI/CD, tools for logging and metrics
  • A continuous drive to explore, improve, enhance, automate and optimize systems and tools.
  • Strong computer science fundamentals in data structures and algorithms
  • Exposure to information retrieval, statistics, and machine learning.
  • Excellent oral and written communication skills.
  • Good analytical and problem-solving skills
At Walmart, we offer competitive pay as well as performance-based bonus awards and other great benefits for a happier mind, body, and wallet. Health benefits include medical, vision and dental coverage. Financial benefits include 401(k), stock purchase and company-paid life insurance. Paid time off benefits include PTO (including sick leave), parental leave, family care leave, bereavement, jury duty, and voting. Other benefits include short-term and long-term disability, company discounts, Military Leave Pay, adoption and surrogacy expense reimbursement, and more. You will also receive PTO and/or PPTO that can be used for vacation, sick leave, holidays, or other purposes. The amount you receive depends on your job classification and length of employment. It will meet or exceed the requirements of paid sick leave laws, where applicable. For information about PTO, see https://one.walmart.com/notices. Live Better U is a Walmart-paid education benefit program for full-time and part-time associates in Walmart and Sam's Club facilities. Programs range from high school completion to bachelor's degrees, including English Language Learning and short-form certificates. Tuition, books, and fees are completely paid for by Walmart.
Eligibility requirements apply to some benefits and may depend on your job classification and length of employment. Benefits are subject to change and may be subject to a specific plan or program terms.
For information about benefits and eligibility, see One.Walmart.
Bentonville, Arkansas US-10735: The annual salary range for this position is $130,000.00 - $260,000.00
Sunnyvale, California US-11807: The annual salary range for this position is $169,000.00 - $338,000.00 Additional compensation includes annual or quarterly performance bonuses. Additional compensation for certain positions may also include :
- StockMinimum Qualifications...

Outlined below are the required minimum qualifications for this position. If none are listed, there are no minimum qualifications.

Option 1: Bachelor's degree in computer science, computer engineering, computer information systems, software engineering, or related area and 4 years' experience in software engineering, machine learning engineering, AI systems or related area.
Option 2: 6 years' experience in software engineering, machine learning engineering, AI systems or related area.Preferred Qualifications...

Outlined below are the optional preferred qualifications for this position. If none are listed, there are no preferred qualifications.

Master's degree in computer science, computer engineering, computer information systems, software engineering, or related area and 2 years' experience in software engineering, machine learning engineering, AI systems or related area., Prior work experience in accessibility best practices., Prior work experience in creating inclusive digital experiences, demonstrating knowledge in implementing Web Content Accessibility Guidelines (WCAG) 2.2 AA standards, assistive technologies, and integrating digital accessibility seamlessly.Primary Location...2501 Se J St, Ste A, Bentonville, AR 72716-3724, United States of AmericaWalmart and its subsidiaries are committed to maintaining a drug-free workplace and has a no tolerance policy regarding the use of illegal drugs and alcohol on the job. This policy applies to all employees and aims to create a safe and productive work environment.

What Walmart employees say

Pay

Benefits

Hours and flexibility

Workplace

Get the full story on Breakroom

Walmart logo

About Walmart

Sourced by ZipRecruiter

From our humble beginnings as a small discount retailer in Rogers, Ark., Walmart has opened thousands of stores in the U.S. and expanded internationally. Through innovation, we're creating a seamless experience to let customers shop anytime and anywhere online and in stores. We are creating opportunities and bringing value to customers and communities around the globe. Walmart operates approximately 10,500 stores and clubs in 19 countries and eCommerce websites. We employ 2.1 million associates around the world — nearly 1.6 million in the U.S. alone.

Industry

Retail, professional, labor and political organizations, specialized design services, transportation and warehousing and fitness and sports centers

Company size

10,000+ Employees

Headquarters location

Bentonville, AR, US

Social media

FacebookTwitter

View All Walmart Jobs

Apply