Searching in torch/distributed