Handles NVIDIA cuda packages.

[yliu120]

🚀 feature request

Relevant Rules

Description

NVIDIA now distributed a bunch of cuda packages through pypi, for instance,

    "nvidia-cublas-cu12==12.8.4.1",
    "nvidia-cuda-cupti-cu12==12.8.90",
    "nvidia-nvjitlink-cu12==12.8.93",
    "nvidia-nvshmem-cu12==3.2.5",

They will be extracted to some shared directory called nvidia. This practice is not usual.

Describe the solution you'd like

I guess we should allow something like this to define a group rule that all the packages in that group will be extracted into a same but not different site-packages folder and the users can attach some bazel rule for this type of external repo.

Describe alternatives you've considered

Implement this locally.

[groodt]

This is a known point of friction.

Closing as duplicate of #2156