--extra-index-url https://download.pytorch.org/whl/cu117/
torch==1.13.0+cu117
torchmetrics==0.10.0
torchvision==0.14.0+cu117
# jax==0.3.0
# jaxlib @ https://storage.googleapis.com/jax-releases/cuda11/jaxlib-0.3.0+cuda11.cudnn805-cp310-none-manylinux2010_x86_64.whl
pytorch-lightning==1.9.4
timm
