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 ↓87% vs prev 7d
total hits 18
actors 6 distinct systems
last hit 4d ago Googlebot
top countries 🇵🇱 Poland · 🇺🇸 United States · 🇩🇪 Germany · 🇫🇷 France · 🇨🇦 Canada
API endpoints
full doc /v1/registry/jax-cuda12-pjrt
compatibility /v1/registry/jax-cuda12-pjrt/compatibility