
Staff Machine Learning Engineer
Walmart
full-time
Posted on:
Location Type: Hybrid
Location: Bentonville • California • United States
Visit company websiteExplore more
Salary
💰 $130,000 - $260,000 per year
Job Level
Tech Stack
About the role
- Develop AI driven intelligent products for the Associate Productivity and Experience team
- Design, architect, build AI/ML models, AI Agents and deploy, operate, optimize the solutions
- Work on Python and mainstream machine learning frameworks, e.g. TensorFlow or PyTorch and Agentic Frameworks like Pydantic
- Lead development of end-to-end system architecture for GenAI/AI/ML and data intensive applications
- Design, 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 telemetry
- Identify or propose AI/ML use-cases to business teams to improve business processes and develop quick MVPs/ POCs
- End to end ownership of product using technologies such as Java, Node JS, React Js
- Review design and code, provide technical solutions to team for complex requirements
- Uphold and enforce rigorous standards, governance, and best practices, ensuring compliance with model and data governance standards.
Requirements
- bachelor's/master's degree in computer science, information technology, engineering, or related area
- Minimum 10 years of experience in design, development, deployment, and maintenance of highly scalable, robust, and user-centric Fullstack/GenAI /AI /ML products and applications
- Good understanding of Data Structures
- Hands 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 Python
- Exposure to cloud infrastructure, such as Open Stack, Azure, GCP, or AWS
- Experience working in Agile Methodology
- Experience using CI/CD, tools for logging and metrics
- A continuous drive to explore, improve, enhance, automate and optimize systems and tools.
- Strong computer science fundamentals in data structures and algorithms
- Exposure to information retrieval, statistics, and machine learning.
- Excellent oral and written communication skills.
- Good analytical and problem-solving skills.
Benefits
- Health benefits include medical, vision and dental coverage.
- 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.
- Tuition, books, and fees are completely paid for by Walmart for education programs.
Applicant Tracking System Keywords
Tip: use these terms in your resume and cover letter to boost ATS matches.
Hard Skills & Tools
PythonAI/ML modelsTensorFlowPyTorchPydanticFull stack developmentJavaNode JSReact JSData Structures
Soft Skills
communication skillsanalytical skillsproblem-solving skillsleadershipcollaboration
Certifications
bachelor's degreemaster's degree