Local CUDA Toolkit Support

[hjnnjh]

Is your feature request related to a problem? Please describe.

I have read the GPU setup guide, which implies that the global CUDA Toolkit is required. However, there is another python environment that contains JAX components in my Linux server, in which the global CUDA Toolkit may cause some unexpected troubles.

Describe the solution you'd like

Is there an alternative that use local CUDA Toolkit(which means installing CUDA Toolkit in a conda virtual environment) instead of a global one?

Describe alternatives you've considered

No response

Additional context

No response

[Notnaton]

@ericrallen Ooba question I believe, we can close this one