(USA) Staff, Machine Learning Engineer
About The Position
The Staff Machine Learning Engineer will lead the design and delivery of scalable, production-grade ML solutions that address critical business needs. This role involves overseeing the end-to-end ML lifecycle, including data sourcing, model development, deployment, and monitoring, while applying MLOps best practices. The engineer will build and maintain ML infrastructure, drive technical standards, and mentor team members. Collaboration across teams and continuous innovation are essential to advance Walmart’s ML capabilities and ensure high-quality, reliable solutions that support business objectives and operational excellence. About the team: The team develops foundational AI models and services that enhance Walmart’s customer-focused applications. Collaborating with Product, Business, Data, and Platform groups, they apply expertise in predictive analytics, machine learning, and data visualization. Key responsibilities include identifying data sources, building scalable models, and ensuring data quality to support informed decisions. Leveraging advanced techniques in natural language processing, computer vision, and deep learning, the team strengthens Walmart’s data strategy and analytics capabilities. They foster continuous learning and innovation to deliver impactful, scalable solutions that drive business value and operational excellence.
Requirements
- Proven expertise in designing and delivering scalable, production-grade machine learning solutions aligned with business needs.
- Strong proficiency in ML frameworks such as TensorFlow, PyTorch, and Scikit-learn, and coding languages including Python and SQL.
- Experience with end-to-end ML lifecycle management, including data sourcing, feature engineering, model training, deployment, and monitoring.
- Knowledge of MLOps best practices, including CI/CD pipelines, model versioning, and telemetry-based monitoring.
- Familiarity with cloud platforms (AWS, GCP, Azure), containerization, Kubernetes, and workflow automation.
- Ability to implement robust model evaluation and validation frameworks.
- Demonstrated leadership in mentoring engineers and enforcing engineering standards.
- Strong collaboration and communication skills to work across teams and stakeholders.
- 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.
Nice To Haves
- 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 accessibility best practices.
- 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.
Responsibilities
- Lead the design and deployment of scalable, production-grade machine learning solutions addressing critical business needs.
- Translate complex problems into clear technical requirements, architectural designs, and measurable success criteria.
- Oversee the full ML lifecycle, including data sourcing, feature engineering, model training, deployment, monitoring, and continuous improvement.
- Implement MLOps best practices such as CI/CD pipelines, model versioning, and monitoring to ensure operational excellence.
- Build and maintain reusable ML infrastructure components and manage large-scale operations across cloud environments using containerization and workflow automation.
- Provide technical leadership and mentorship, setting standards for code quality, testing, and deployment.
- Drive experimentation pipelines and explore advanced ML techniques to accelerate innovation and solution delivery.
Benefits
- Performance-based bonus awards
- Medical coverage
- Vision coverage
- Dental coverage
- 401(k)
- Stock purchase
- Company-paid life insurance
- PTO (including sick leave)
- Parental leave
- Family care leave
- Bereavement
- Jury duty
- Voting
- Short-term disability
- Long-term disability
- Company discounts
- Military Leave Pay
- Adoption and surrogacy expense reimbursement
- PTO and/or PPTO that can be used for vacation, sick leave, holidays, or other purposes
- Live Better U (Walmart-paid education benefit program for full-time and part-time associates)
Stand Out From the Crowd
Upload your resume and get instant feedback on how well it matches this job.
What This Job Offers
Job Type
Full-time
Career Level
Senior
