Job Utilities#

get_rank([default])

Return the process rank obtained from a supported environment variable.

get_task_index()

Return the task index for the current array job.

get_task_stepsize()

Return the task step size for the current array job.

split_tasks(task_list)

Return the task shard assigned to the current array job.