maxent_grpo.training.rollout.distributed

Distributed helpers shared across generation utilities.

Functions

_broadcast_object_list(accelerator, payload, *)

Broadcast python objects even when Accelerate lacks the helper.

_gather_object_list(accelerator, value)

Gather Python lists across ranks with graceful Accelerate fallbacks.

_scatter_object(accelerator, input_list, *)

Scatter python objects from src to all ranks.