extra credit Search 85,062 technologies ⌘K Download for Mac

jax-cuda13-pjrt

JAX XLA PJRT Plugin for NVIDIA GPUs

pip install jax-cuda13-pjrt
↑↓ move ↵ open