Staff, Machine Learning Engineer

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. Execution of multiple business plans and projects: identifies customer and operational needs; develops and communicates business plans and priorities; removes barriers and obstacles that impact performance; provides resources; identifies performance standards; measures progress and adjusts performance accordingly; develops contingency plans; and demonstrates adaptability and supports continuous learning. Supports company policies, procedures, mission, values, and standards of ethics and integrity by training and providing direction to others in their use and application; ensures compliance with them; and utilizes and supports 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. 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 a 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. 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). 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. 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. 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. Seeks and implements continuous improvements and encourages the team to leverage new digital tools and ways of working. Benefits 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. 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 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. 6 years’ experience in software engineering, machine learning engineering, AI systems or related area. 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. 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, abilities, ideas and opinions- while being inclusive of all people. #J-18808-Ljbffr