Apply on Employer Site

Walmart Canada · 2 months ago

Staff, Machine Learning Engineer

United States

Full-time

Remote

Senior Level, Lead/Staff

$169K/yr - $338K/yr

4+ years exp

Walmart Inc. is a leading retail corporation, and they are seeking a Staff Machine Learning Engineer. The role involves leading the design and implementation of scalable ML solutions, overseeing the entire ML lifecycle, and providing technical leadership and mentorship to engineers.

DeliveryRetailShopping

Responsibilities

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

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

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

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

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

Demonstrates responsibility for deliverables, ensuring solutions meet quality standards and business objectives. Takes initiative to resolve issues and drive projects to completion

Applies sound reasoning to make informed technical decisions, balancing trade-offs and considering long-term impacts on system architecture and business goals

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

Works effectively across teams, openly shares knowledge and communicates complex ideas clearly to both technical and non-technical stakeholders

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

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

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

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

Acts as an altruistic servant leader and is consistently humble, self-aware, honest, and transparent

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

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

Qualification

Machine Learning EngineeringML FrameworksCloud EnvironmentsMLOps Best PracticesPythonSQLTechnical LeadershipChange AgilityContinuous ImprovementInnovationCollaboration

Required

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

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

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

Knowledge of accessibility best practices and Walmart's accessibility standards and guidelines for supporting an inclusive culture

Benefits

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.

Live Better U is a Walmart-paid education benefit program for full-time and part-time associates in Walmart and Sam's Club facilities.