Staff, Machine Learning Engineer

Walmart · Retail · Bentonville, AR +1

Staff Machine Learning Engineer to develop AI-driven intelligent products for Associate Productivity and Experience. Responsibilities include designing, building, and deploying AI/ML models and agents, leading end-to-end system architecture for GenAI/AI/ML applications, and developing production-grade ML services. Requires strong experience in full-stack development, Python, ML frameworks, and cloud infrastructure.

What you'd actually do

  1. Design, architect, build AI/ML models, AI Agents and deploy, operate, optimize the solutions
  2. Work on Python and mainstream machine learning frameworks, e.g. TensorFlow or PyTorch and Agentic Frameworks like Pydantic
  3. Lead development of end-to-end system architecture for GenAI/AI/ML and data intensive applications
  4. Design, Develop and deploy Full stack based applications.
  5. 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

Skills

Required

  • Python
  • TensorFlow or PyTorch
  • Agentic Frameworks like Pydantic
  • Java
  • Node JS
  • React Js
  • Cloud infrastructure (Open Stack, Azure, GCP, or AWS)
  • CI/CD
  • logging and metrics tools
  • Data Structures
  • Algorithms
  • Distributed systems
  • System design

Nice to have

  • Full stack developer
  • Information retrieval
  • Statistics
  • Machine learning

What the JD emphasized

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

Other signals

  • Develop AI driven intelligent products
  • Design, architect, build AI/ML models, AI Agents
  • Lead development of end-to-end system architecture for GenAI/AI/ML
  • Develop and deploy production-grade real-time/batch machine learning services
  • Industry level experience in developing production-grade AI/ML system development