Update instructions for ROCM

[weaponmasterjax]

Pull request checklist

• [y] The PR has a proper title. Use Semantic Commit Messages. (No more branch-name title please)
• [y] Make sure you are requesting the right branch: dev.
• [y] Make sure this is ready to be merged into the relevant branch. Please don't create a PR and let it hang for a few days.
• [y] Ensure all tests are passing.
• [y] Ensure linting is passing.

PR type

• chore

Description

• Update instructions for ROCM since ROCM now fully supports PyTorch, and it is confirmed that a drop-in replacement of PyTorch to the ROCM version will work. This will enable AMD GPU acceleration in more scenarios.
• Do to the complexity of dependencies it's easier to just replace the existing CUDA torch installation with ROCM ones afterwards, rather than doing that again after PyTorch being overwritten in the process of pip install -r requirements.txt .

[fumiama]

Thanks!