examples/research_projects/rag/distributed_pytorch_retriever.py/RagPyTorchDistributedRetriever/_is_main
class RagPyTorchDistributedRetriever:    def _is_main(self):
        return dist.get_rank(group=self.process_group) == 0