gigl.common.utils.get_rank

gigl.common.utils.vertex_ai_context.get_rank() → int

Rank of the current VAI process, so they will know whether it is the master or a worker. Note: that VAI only creates one process per machine. It is the user’s responsibility to create multiple processes per machine. Meaning, this function will only return one integer for the main process that VAI creates. Throws if not on Vertex AI.