torch
numpy
fsspec
tensorboard
psutil
typing_extensions
setuptools

[dev]
pytest
pytest-cov
