gigl.common.utils.get_world_size

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

The total number of processes that VAI creates. Note that VAI only creates one process per machine. It is the user’s responsibility to create multiple processes per machine. Throws if not on Vertex AI.