pytorch setup with uv

configuring pytorch with uv package manager for different compute backends