Senior AI/ML Engineer with Walmart Codebase || Remote

Title: Senior AI/ML Engineer with Walmart Codebase

Location: Remote

Job Type: Contract/Fulltime

Senior Software Engineer who brings deep knowledge of Walmart’s engineering ecosystem along with hands-on AI/ML expertise. This role blends robust backend development with data-driven intelligence to power scalable, personalized retail experiences.

Responsibilities

• Architect, build, and deploy AI-enhanced features such as product recommendations, intelligent search, and dynamic pricing integrated with Walmart’s existing platfoJrms.

• Design and integrate ML models into backend microservices and frontend experiences, ensuring real-time performance and scalability.

• Collaborate with cross-functional teams, including data science, MLOps, and retail analytics, to drive customer-centric solutions.

• Leverage internal Walmart tools and frameworks for model training, deployment, and monitoring within the company’s CI/CD pipeline.

• Maintain code quality and AI system reliability using test automation, canary deployments, and continuous monitoring.

• Participate in AI/ML experimentation frameworks, A/B testing, and inference optimization at scale.

Required Skills

• Experience working with Walmart codebase or previous roles at Walmart Labs / Walmart Global Tech.

• Proficient in Java, Node.js, React/Redux, Spring Boot, and Walmart’s microservice architecture.

• Strong applied knowledge in AI/ML with tools like TensorFlow, PyTorch, scikit-learn, or XGBoost.

• Experience deploying models using REST APIs, ONNX, or TensorFlow Serving.

• Hands-on with data pipelines, feature engineering, and training data curation using Python, Pandas, Spark, or BigQuery.

• Familiarity with AI/ML use cases in retail: recommendations, customer segmentation, demand forecasting, and computer vision for product tagging or in-store analytics.

• Cloud experience with GCP, AWS SageMaker, or internal Walmart AI infrastructure.

• Understanding of Walmart's AI deployment practices, CI/CD tools (e.g., OneOps, Jenkins), and experimentation platforms.

Education

• Bachelor's or Master’s degree in Computer Science, Engineering, or a related field.