gigl.common.utils.vertex_ai_context

Utility functions to be used by machines running on Vertex AI.

Functions

connect_worker_pool	Used to connect the worker pool.
get_host_name	Get the current machines hostname.
get_leader_hostname	Hostname of the machine that will host the process with rank 0.
get_leader_port	A free port on the machine that will host the process with rank 0.
get_rank	Rank of the current VAI process, so they will know whether it is the master or a worker.
get_vertex_ai_job_id	Get the Vertex AI job ID.
get_world_size	The total number of processes that VAI creates.
is_currently_running_in_vertex_ai_job	Check if the code is running in a Vertex AI job.