Getting the right version of PyTorch installed to have the correct kind of acceleration on each different platform you support has been a long-standing headache across many Python dependency management tools, not just uv. For example, here's the bug in poetry regarding this issue:
https://github.com/python-poetry/poetry/issues/6409As I understand it, recent versions of PyTorch have made this process somewhat easier, so maybe it's worth another try.