label_smoothing: float = 0.0,
) -> Tensor:
"""Distributed cross_entropy entry used by shard dispatch."""
# Defer the components import to preserve the lightweight models import boundary.
from hyper_parallel.components.losses._vocab_parallel_cross_entropy import ( # pylint: disable=C0415
DistributedCrossEntropyFunction,
)
input_dtensor = None