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

jax-cuda12-pjrt

JAX XLA PJRT Plugin for NVIDIA GPUs

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