Demo

Staff, Machine Learning Engineer

Walmart Global Tech
Sunnyvale, CA Full Time
POSTED ON 3/27/2025
AVAILABLE BEFORE 4/20/2025

As a company, Walmart primarily generates revenue through advertising, and we're working towards building a massive business to fund our mission. The Display Ads ML team builds services that enable advertisers to efficiently capture and predict user engagement while minimizing their costs. The team leverages machine learning models to incorporate advertisers’ goals in auction ranking, allowing them to maximize installs, key conversions, or return on ad spend. As the primary owners of the engagement prediction and bid recommendation systems, the team is responsible for a central part of the ads delivery system that generates bid recommendation considering advertiser objectives and user engagement scores. Given the centrality of this system, the team also regularly conducts highly rigorous and controlled experiments to measure the impact of improvements to these systems. Some of the project highlights include maximizing conversions campaign optimization, which uses real-time data to meet advertisers’ key objectives.

As a Staff Machine Learning Engineer, you will work closely with technical leaders to build new advertiser optimization products that significantly increase the efficiency and ROI for the advertisers.

  • You will be responsible for the quality and technical approach within the team; partner with other leads in direction setting, planning, and overseeing engineering designs and executions; establish and contribute to the group’s culture and processes.
  • You will collaborate with cross-functional engineering, product, and marketing teams to ensure that our customers’ voices are represented in your work. You will design, conduct, and analyze experiments to quantify the impact of the systems that you build.
  • We are seeking a highly skilled engineer to take a critical role in building state-of-the-art bidding optimization and user prediction systems. As a leader of this team, you will help us uplevel the capabilities of Walmart Display Ads platform through innovative optimization products that enhance advertiser value.

About Team :

Our team works closely with our US stores and eCommerce business to better serve customers by empowering team members, stores, and merchants with technological innovation. From groceries and entertainment to sporting goods and crafts, Walmart U.S. offers an extensive selection that our customers value, whether they shop online at Walmart.com, through one of our mobile apps, or in-store. Focus areas include customers, stores and employees, in-store service, merchant tools, merchant data science, and search and personalization.

What you'll do :

  • Set Technical Vision and Strategy : Define and execute a roadmap for engagement and bid recommendation modeling, balancing innovative modeling approaches with business objectives. Build Walmart-scale optimizations to improve advertiser outcomes using cutting-edge techniques in the industry.
  • Drive Technical Execution : Oversee the model development lifecycle from ideation to deployment, ensuring high standards of ML performance and robustness.
  • Lead and Mentor a High-Performing Team : Recruit, mentor, and retain top ML talent, fostering a culture of growth, collaboration, and technical excellence. Lead the team on designing new user engagement and bid optimization products and algorithms as well as conducting rigorous A / B experiments to evaluate the business impact.
  • Collaborate Cross-Functionally : Partner with PMs, data scientists, and other engineering teams to align on engagement strategies, data requirements, and model KPIs.
  • Innovate in ML Architecture : Implement and optimize model architectures tailored to engagement prediction, leveraging deep learning and advanced ML techniques. Think out-of-the box!
  • What you'll bring :

  • 10 years of contributing high-quality code to production systems that operate at scale.
  • Lead / participate the design, development and implementation of complex machine learning solutions / algorithms that drive business results leveraging Python, Nodejs, LLM, NLP Models, Chat GPT or similar platforms, scikit-learn, TensorFlow / Pytorch, Cassandra, BQ, Redis
  • Solved complex problems with multilayered data sets, multi batch inferencing. Upholding ML engineering best practices - enforcing high standard for quality, reliability, and security in deployed machine learning solutions.
  • Stay up-to-date with the latest AI / ML technologies and trend - acting as a thought leader for ML Engineering within the organization and in the broader technical community.
  • Experience designing optimization algorithms in an ad serving platform and / or other marketplaces is preferred.
  • About Walmart Global Tech

    Imagine working in an environment where one line of code can make life easier for hundreds of millions of people. That’s what we do at Walmart Global Tech. We’re a team of software engineers, data scientists, cybersecurity expert's and service professionals within the world’s leading retailer who make an epic impact and are at the forefront of the next retail disruption. People are why we innovate, and people power our innovations. We are people-led and tech-empowered. We train our team in the skillsets of the future and bring in experts like you to help us grow. We have roles for those chasing their first opportunity as well as those looking for the opportunity that will define their career. Here, you can kickstart a great career in tech, gain new skills and experience for virtually every industry, or leverage your expertise to innovate at scale, impact millions and reimagine the future of retail.

    Walmart’s culture is a competitive advantage, and it’s fostered by being together. Working together in person allows us to collaborate, align quickly and innovate with greater speed. We use our campuses to create purposeful connection rooted in deepening understanding and investing in the development of our associates.

    Our hubs : Walmart is a global company with offices across the United States and around the world. Our global headquarters is in Bentonville, Arkansas, with primary hubs in the San Francisco Bay area and New York / New Jersey.

    Benefits :

    Benefits : Beyond our great compensation package, you can receive incentive awards for your performance. Other great perks include 401(k) match, stock purchase plan, paid maternity and parental leave, PTO, multiple health plans, and much more.

    Equal Opportunity Employer :

    Walmart, Inc. is an Equal Opportunity Employer – By Choice. We believe we are best equipped to help our associates, customers and the communities we serve live better when we really know them. That means understanding, respecting and valuing diversity- unique styles, experiences, identities, ideas and opinions – while being inclusive of all people.

    The above information has been designed to indicate the general nature and level of work performed in the role. It is not designed to contain or be interpreted as a comprehensive inventory of all responsibilities and qualifications required of employees assigned to this job. The full Job Description can be made available as part of the hiring process.

    If your compensation planning software is too rigid to deploy winning incentive strategies, it’s time to find an adaptable solution. Compensation Planning
    Enhance your organization's compensation strategy with salary data sets that HR and team managers can use to pay your staff right. Surveys & Data Sets

    What is the career path for a Staff, Machine Learning Engineer?

    Sign up to receive alerts about other jobs on the Staff, Machine Learning Engineer career path by checking the boxes next to the positions that interest you.
    Income Estimation: 
    $119,030 - $151,900
    Income Estimation: 
    $149,493 - $192,976
    Income Estimation: 
    $119,030 - $151,900
    Income Estimation: 
    $149,493 - $192,976
    Income Estimation: 
    $77,900 - $95,589
    Income Estimation: 
    $101,387 - $124,118
    Income Estimation: 
    $101,387 - $124,118
    Income Estimation: 
    $119,030 - $151,900
    Income Estimation: 
    $149,493 - $192,976
    Income Estimation: 
    $184,796 - $233,226
    View Core, Job Family, and Industry Job Skills and Competency Data for more than 15,000 Job Titles Skills Library

    Job openings at Walmart Global Tech

    Walmart Global Tech
    Hired Organization Address Sunnyvale, CA Full Time
    ** Please note that Immigration Sponsorship is not available for this role .** We are looking for an experienced Princip...
    Walmart Global Tech
    Hired Organization Address Sunnyvale, CA Full Time
    Immigration Sponsorship is not available in this role & location About Team Marketplace Engineering team is at the foref...
    Walmart Global Tech
    Hired Organization Address Sunnyvale, CA Full Time
    Immigration sponsorship is not available in this role. This role involves collaboration with multiple stakeholders to im...
    Walmart Global Tech
    Hired Organization Address Sunnyvale, CA Full Time
    Join Walmart as Principal Data Scientist in our Walmart Transactional Systems team. As a Principal Data Scientist, you w...

    Not the job you're looking for? Here are some other Staff, Machine Learning Engineer jobs in the Sunnyvale, CA area that may be a better fit.

    Staff Machine Learning Engineer

    Candidate Experience site, Sunnyvale, CA

    Staff Machine Learning Engineer

    Visa, Santa Clara, CA

    AI Assistant is available now!

    Feel free to start your new journey!