pytorch

771a5ae8 - Merge branch 'master' into lazy_tensor_staging