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

[train]
datasets
gcsfs
wandb
