Dharmendra04

Hi, I tried your way and installed jax gpu version using pip install --upgrade "jax[cuda12_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html, and ran the below codes, from jax.lib import xla_bridge print(xla_bridge.get_backend().platform), it is showing,...

https://onedrive.live.com/?authkey=%21AGNgYRrq6OQPZE4&id=1B5D487B5A28CEFD%213218&cid=1B5D487B5A28CEFD you can find it here