pytorch

6abf1513 - update to one allgather operation