torch
numpy
accelerate
astropy
humanize
psutil
GPUtil
nflows
huggingface_hub
datasets>=4.0
pyarrow
fsspec
