Reduce memory by using all_gather_into_tensor
#3277
This run and associated checks have been archived and are scheduled for deletion.
Learn more about checks retention