torch
numpy
fsspec
tensorboard
packaging
psutil
pyre_extensions
typing_extensions
setuptools
tqdm
tabulate

[dev]
parameterized
pytest
pytest-cov
torchsnapshot-nightly
pyre-check
torchvision==0.16.2
