pub trait DistributedOperation { // Required method fn sum_all_reduce(&self, xs: &Tensor) -> Result<Tensor>; }