"""Distributed training helpers."""