Distributed training and scalability for the particle clustering method UCluster