click
pyyaml
toml
pytest
torch
lightning
scipy
numpy
xarray
rich
tensorboard
axial-attention
torchvision
