perf(aggregation): Prevent cuda sync in normalize#557

ValerianRey merged 2 commits intomainSimplexLab/TorchJD:mainfrom

prevent-cuda-sync-normalizeSimplexLab/TorchJD:prevent-cuda-sync-normalizeCopy head branch name to clipboard