https://us-python.pkg.dev/ml-oss-artifacts-published/jax-public-nightly-artifacts-registry/simple/jax-cuda12-pjrt
Links for jax-cuda12-pjrt
links forjax
https://openxla.org/xla/pjrt
PJRT - Uniform Device API | OpenXLA Project
device apiuniformproject