Sr Staff, Machine Learning Engineer

Position Summary...

What you'll do...

ML System Architecture & Delivery: •Lead the design and implementation of scalable, production-grade ML solutions that address business-critical needs. •Translate complex or ambiguous problem statements into clear technical requirements, architectural designs, and success metrics. •Select and integrate appropriate ML frameworks (Scikit-learn, TensorFlow, PyTorch, XGBoost, Spark ML) and coding languages (Python, SQL, Java, C++, R) to meet performance, explainability, and maintainability standards. •Drive platform alignment by ensuring solutions leverage and contribute to Walmart’s shared ML infrastructure and tooling. End-to-End ML Lifecycle Execution: •Oversee and contribute to all phases of the ML lifecycle—data sourcing, feature engineering, model training, deployment, monitoring, and continuous improvement. •Apply MLOps best practices such as CI/CD for ML, automated training pipelines, model versioning, and telemetry-based monitoring. •Implement robust evaluation frameworks for model performance, data quality, and fairness, ensuring compliance with responsible AI principles. ML Platform Engineering & Operations: •Build and maintain reusable infrastructure components, including feature stores, model runtimes, SDKs, and workflow orchestration systems. •Manage large-scale ML operations across cloud environments (AWS, GCP, Azure), leveraging containerization, Kubernetes, and workflow automation. •Implement observability tooling to track model behavior, detect drift, and ensure high availability and low latency in production environments. Technical Leadership & Mentorship: •Provide hands-on mentorship to engineers, guiding them through architectural decisions, implementation details, and debugging complex ML issues. •Set and enforce engineering standards for code quality, testing, deployment, and operational excellence. •Foster a culture of technical depth, peer learning, and collaboration across teams. Experimentation & Innovation: •Design and manage experimentation pipelines for hypothesis testing, A/B experiments, and continuous model iteration. •Explore advanced techniques such as AutoML, neural architecture search, domain-specific NLP tooling, and multimodal architectures to accelerate solution delivery. •Share innovations and learnings through technical documentation, presentations, and cross-team forums to elevate Walmart’s ML engineering practices. Technical Ownership: •Demonstrates responsibility for deliverables, ensuring solutions meet quality standards and business objectives. Takes initiative to resolve issues and drive projects to completion. Technical Judgment: •Applies sound reasoning to make informed technical decisions, balancing trade-offs and considering long-term impacts on system architecture and business goals. Change Agility: •Adapts quickly to change, embraces new technologies, and understands the broader system context. Anticipates downstream effects of technical choices and integrates feedback to improve outcomes. Collaboration: •Works effectively across teams, openly shares knowledge and communicates complex ideas clearly to both technical and non-technical stakeholders. Innovation & Continuous Improvement: •Proactively seeks opportunities to enhance processes, tools, and products. Encourages experimentation and learning to drive ongoing improvement. Drives the execution of multiple business plans and projects by identifying customer and operational needs; developing and communicating business plans and priorities; removing barriers and obstacles that impact performance; providing resources; identifying performance standards; measuring progress and adjusting performance accordingly; developing contingency plans; and demonstrating adaptability and supporting continuous learning. Promotes and supports company policies, procedures, mission, values, and standards of ethics and integrity by training and providing direction to others in their use and application; ensuring compliance with them; and utilizing and supporting the Open Door Policy. Ensures business needs are being met by evaluating the ongoing effectiveness of current plans, programs, and initiatives; consulting with business partners, managers, co-workers, or other key stakeholders; soliciting, evaluating, and applying suggestions for improving efficiency and cost effectiveness; and participating in and supporting community outreach events. Provides supervision and development opportunities for associates by selecting and training; mentoring; assigning duties; building a team-based work environment; establishing performance expectations and conducting regular performance evaluations; providing recognition and rewards; coaching for success and improvement; and promoting a belonging mindset in the workplace. Provides supervision and development opportunities for associates by selecting and training; mentoring; assigning duties; building a team-based work environment; establishing performance expectations and conducting regular performance evaluations; providing recognition and rewards; coaching for success and improvement; and promoting a belonging mindset in workplace. Respect the Individual: Demonstrates and encourages respect for all; builds a high-performing team; seeks, and embraces differences in people, cultures, ideas and experiences; creates a workplace where all associates feel seen, supported and connected through culture of belonging so associates thrive and perform; drives a positive associate and customer/member experience for all; identifies, attracts, and retains the best team members. Respect the Individual: Creates a discipline and focus around developing talent, through feedback, coaching, mentoring, and developmental opportunities; promotes an environment allowing everyone to bring their best selves to work; empowers associates and partners to act in the best interest of the customer/member and company; and regularly recognizes others’ contributions and accomplishments. Respect the Individual: Builds strong and trusting relationships with team members and business partners; works collaboratively and cross-functionally to achieve objectives; and communicates and listens attentively, with energy and positivity to motivate, influence, and inspire commitment and action. Act with Integrity: Maintains and promotes the highest standards of integrity, ethics and compliance; models the Walmart values and leads by example to foster our culture; supports Walmart’s goal of becoming a regenerative company by making a positive impact for associates, customers, and the world around us (e.g., creating a sense of belonging, eliminating waste, participating in local giving). Act with Integrity: Follows the law, our code of conduct and company policies, and sets expectations for others to do the same; promotes an environment where associates feel comfortable sharing concerns and reinforces our culture of non-retaliation; listens to concerns raised by associates. takes action and encourages others to do the same; holds self and others accountable for achieving results in a way that is consistent with our values. Act with Integrity: Acts as an altruistic servant leader and is consistently humble, self-aware, honest, and transparent. Serve our Customers and Members: Delivers expected business results while putting the customer/member first and consistently applying an omni-merchant mindset and acts with an Every Day Low-Cost mindset to drive value and Every Day Low Prices for customers/members. Serve our Customers and Members: Adopts a holistic perspective that considers data, analytics, customer/member insights, and different parts of the business when making plans and shaping the team’s strategy. Strive for Excellence: Consistently raises the bar and seeks to improve; demonstrates curiosity and a growth mindset; seeks feedback, asks thoughtful questions, fosters an environment that supports learning, innovation, and learning from mistakes, and intelligent risk-taking; and exhibits resilience in the face of setbacks. Strive for Excellence: Seeks and implements continuous improvements and encourages the team to leverage new digital tools and ways of working. 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.
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 :
- Stock

ㅤ

ㅤ

ㅤ

ㅤ

‎ 

Minimum 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., We value candidates with a background 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. The ideal candidate would have knowledge of accessibility best practices and join us as we continue to create accessible products and services following Walmart’s accessibility standards and guidelines for supporting an inclusive culture.

Primary Location...

1395 Crossman Ave, Sunnyvale, CA 94089-1114, United States of America

Walmart 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.