yay -S python-pytorch-rocm python-torchvision-rocm python-torchaudio-rocm
#uv pip install --pre torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3