JAX CUDA 12 PJRT Plugin
JSON →The jax-cuda12-pjrt package provides the JAX XLA PJRT backend for NVIDIA GPUs, specifically built with CUDA 12. It serves as the `jaxlib` implementation when GPU acceleration is desired. The current version is 0.9.2, and JAX along with its ecosystem components typically follow a rapid release cadence, often with monthly or bi-monthly updates.
Traffic · last 30 days stale · no recent hits
total hits 10
actors 4 distinct systems
last hit 16d ago AhrefsBot
top countries 🇺🇸 United States · 🇨🇦 Canada · 🇩🇪 Germany · 🇫🇷 France
API endpoints
full doc /v1/registry/jax-cuda12-pjrt
compatibility /v1/registry/jax-cuda12-pjrt/compatibility