Walmart says AI training is boosting confidence with the technology among its 2.1 million associates