torch
torchvision
einops
numpy
scipy
h5py
matplotlib
pandas
tqdm
fsspec
obspy
gcsfs
datasets
pyarrow
wandb
scikit-learn
