{"categories":[{"label":"Artificial Intelligence","url":"https://skillfed.io/packages/category/scientific-engineering-artificial-intelligence/4"}],"enrichment":{"capability":"Provides NVIDIA GPU acceleration for JAX numerical computing via the PJRT plugin, enabling compiled array operations and automatic differentiation on CUDA 12 hardware.","skillfed_tags":["gpu-acceleration","jax-plugin","cuda"],"use_cases":["Training deep learning models on NVIDIA GPUs using JAX's automatic differentiation and JIT compilation.","Running large-scale numerical simulations and scientific computing workloads with GPU acceleration.","Developing machine learning research code that needs efficient gradient computation across GPU clusters.","Accelerating array operations and linear algebra computations on NVIDIA hardware without writing CUDA code.","Deploying JAX-based inference pipelines that require GPU performance for production inference."],"what_it_does":"jax-cuda12-pjrt is a plugin that connects JAX's XLA compiler to NVIDIA GPUs via the PJRT (Portable Runtime) interface, enabling GPU-accelerated execution of JAX programs on CUDA 12 hardware. It allows you to write numerical code using JAX's NumPy-like API and automatic differentiation, then compile and run it efficiently on NVIDIA accelerators.\n\nThe package is a specialized distribution component of JAX itself\u2014it bundles the PJRT runtime plugin for CUDA 12 in a platform-specific wheel. Installation brings GPU support to JAX's core transformations (grad, jit, vmap) without requiring manual compilation of the plugin. It is actively maintained and carries no known security vulnerabilities.","worth_installing":"Yes, if you are using JAX on an NVIDIA GPU with CUDA 12 and need GPU acceleration. The plugin is actively maintained, carries no known vulnerabilities, and is essential for GPU-backed JAX workloads. Install friction is moderate due to platform specificity and CUDA 12 runtime dependency, but this is expected for GPU-accelerated packages. Not needed if you are using CPU-only JAX or a different GPU backend."},"id":"jax-cuda12-pjrt","links":{"html":"https://skillfed.io/packages/jax-cuda12-pjrt","md":"https://skillfed.io/packages/jax-cuda12-pjrt.md","pypi":"https://pypi.org/project/jax-cuda12-pjrt/"},"maintenance":{"status":"active"},"meta":{"latest_release":"2026-07-16","license_spdx":null,"license_treatment":"permissive","name":"jax-cuda12-pjrt","python_support":"unspecified","summary":"JAX XLA PJRT Plugin for NVIDIA GPUs"},"popularity":{"monthly_downloads":774416,"position":5095,"tier":"top_15000"},"security":{"n_vulnerabilities":0},"version":"0.11.0"}