Deprecate `DistContext`

[mkolodner-sc]

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.