Staff, Machine Learning Engineer

Reference: R-2395925Posted: 2026-01-16Location: BENTONVILLE, AR, United States; Sunnyvale, CASalary: $130000 - $260000/yearType: Regular/PermanentCompany: WalmartWhat 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 solutionsWork on Python and mainstream machine learning frameworks, e.g. TensorFlow or PyTorch and Agentic Frameworks like PydanticLead development of end-to-end system architecture for GenAI/AI/ML and data intensive applicationsDesign, 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 telemetryIdentify or propose AI/ML use-cases to business teams to improve business processes and develop quick MVPs/ POCs to help stakeholders make better decisionsEnd to end ownership of product using technologies such as Java , Node JS, React JsYou will be required to review design and code, provide technical solutions to team for complex requirementsUphold 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 applicationsWhat you'll bring:bachelor's/master's degree in computer science, information technology, engineering, or related areaMinimum 10 years of experience in design, development, deployment, and maintenance of highly scalable, robust, and user-centric Fullstack/GenAI /AI /ML products and applicationsGood understanding of Data StructuresHands 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 PythonExposure to cloud infrastructure, such as Open Stack, Azure, GCP, or AWSExperience working in Agile MethodologyExperience using CI/CD, tools for logging and metricsA continuous drive to explore, improve, enhance, automate and optimize systems and tools.Strong computer science fundamentals in data structures and algorithmsExposure 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 . 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["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["Prior work experience in accessibility best practices.", "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 creating inclusive digital experiences, demonstrating knowledge in implementing Web Content Accessibility Guidelines (WCAG) 2.2 AA standards, assistive technologies, and integrating digital accessibility seamlessly."]