torchscale/examples/fairseq/tasks/data/__init__.py