Deprecate `DistContext`

We should fully deprecate our use of `DistContext` and fully rely on the distributed process groups to pass around ranks/world_sizes in our dataloader, training, and inference code. This is making our Dataloaders more complicated than necessary and blocking some simplification of other APIs, such as our `init_neighbor_loader_worker` function.