AI for warehouse management