ScheduleLoopedBFS

class tensorplay.distributed.pipelining.ScheduleLoopedBFS(stages: list[Any], n_microbatches: int, loss_fn: Any = None, output_merge_spec: Any = None, scale_grads: bool = True, backward_requires_autograd: bool = True, defer_pp_recv: bool = False, max_active_stages: int | None = None)[source]