{"categories":[{"label":"Scientific/Engineering","url":"https://skillfed.io/packages/category/scientific-engineering/4"}],"enrichment":{"capability":"Provides NVIDIA GPU acceleration for JAX numerical computing via the PJRT plugin interface, enabling XLA compilation and execution on CUDA 13 hardware.","skillfed_tags":["gpu-acceleration","machine-learning","numerical-computing"],"use_cases":["Train neural networks on NVIDIA GPUs using JAX's automatic differentiation and JIT compilation.","Accelerate scientific computing workloads (linear algebra, FFTs, stochastic simulations) on GPU hardware.","Run JAX-based machine learning research code on NVIDIA clusters or cloud instances with CUDA 13.","Develop high-performance numerical applications requiring GPU parallelism using vmap."],"what_it_does":"This package is a PJRT (Portable Runtime for JAX) plugin that bridges JAX's numerical computing framework to NVIDIA GPUs via CUDA 13. It enables JAX's core transformations\u2014automatic differentiation (grad), JIT compilation, and vectorization (vmap)\u2014to execute on NVIDIA hardware by interfacing with XLA's compiler infrastructure.\n\nWhen installed, it allows JAX programs to transparently offload array computations and neural network training to NVIDIA GPUs without code changes. The plugin handles device memory management, kernel dispatch, and multi-device coordination. It is a platform-specific binary distribution (wheels for Linux x86_64 and aarch64 only) that must match your system's CUDA 13 installation.","worth_installing":"Yes, if you have NVIDIA GPU hardware with CUDA 13 and want to use JAX for accelerated computing. The package is actively maintained, permissively licensed, and has no known vulnerabilities. Install friction is moderate due to platform specificity and CUDA system dependency, but necessary for GPU support. Not applicable on CPU-only or non-NVIDIA systems."},"id":"jax-cuda13-pjrt","links":{"html":"https://skillfed.io/packages/jax-cuda13-pjrt","md":"https://skillfed.io/packages/jax-cuda13-pjrt.md","pypi":"https://pypi.org/project/jax-cuda13-pjrt/"},"maintenance":{"status":"active"},"meta":{"latest_release":"2026-07-16","license_spdx":null,"license_treatment":"permissive","name":"jax-cuda13-pjrt","python_support":"unspecified","summary":"JAX XLA PJRT Plugin for NVIDIA GPUs"},"popularity":{"monthly_downloads":415248,"position":6827,"tier":"top_15000"},"security":{"n_vulnerabilities":0},"version":"0.11.0"}