--extra-index-url https://download.pytorch.org/whl/cu128 torch==2.7.0+cu128 torchvision click numpy einops omegaconf tqdm rerun-sdk imageio[ffmpeg] opencv-python matplotlib python-pycg timm transformers gdown OpenEXR<3.3.0 kornia hydra-core ray ninja viser