fairseq/tests/gpu/__init__.py