tensor-group-sym / python / large_scale / starg_torch