{"categories":[{"label":"Artificial Intelligence","url":"https://skillfed.io/packages/category/scientific-engineering-artificial-intelligence/5"}],"enrichment":{"capability":"Provides NVIDIA GPU support for JAX by enabling CUDA 13 compilation and execution on compatible GPUs through the jax-cuda13-pjrt runtime.","skillfed_tags":["gpu-acceleration","cuda-plugin","machine-learning"],"use_cases":["Accelerate machine learning training on NVIDIA GPUs by enabling JAX's JIT compilation and automatic differentiation to target GPU hardware.","Run numerical simulations and scientific computations that benefit from GPU parallelization through JAX's vmap and pmap transformations.","Deploy JAX-based inference pipelines on GPU clusters for production machine learning systems.","Develop and test GPU-accelerated JAX code on Linux systems with NVIDIA GPUs before scaling to multi-GPU environments."],"what_it_does":"jax-cuda13-plugin is a platform-specific plugin that enables JAX to compile and execute numerical computations on NVIDIA GPUs using CUDA 13. It acts as a bridge between JAX's compiler infrastructure and NVIDIA's GPU hardware, allowing automatic differentiation, JIT compilation, and vectorization operations to run on GPUs for significant performance gains in machine learning and scientific computing workloads.\n\nThe plugin is tightly integrated with JAX's ecosystem and depends on jax-cuda13-pjrt for the underlying PJRT implementation. It is distributed as pre-compiled wheels for Linux systems (both x86_64 and aarch64 architectures) and supports Python 3.12, 3.13, and 3.14, including free-threaded variants. Installation requires CUDA 13 to be present on the system.","worth_installing":"Yes, if you are using JAX on Linux with an NVIDIA GPU and CUDA 13 installed. This plugin is essential for GPU acceleration and is actively maintained with no known vulnerabilities. Install friction is moderate but manageable for systems meeting the platform requirements. Not applicable on non-Linux systems or without compatible NVIDIA hardware."},"id":"jax-cuda13-plugin","links":{"html":"https://skillfed.io/packages/jax-cuda13-plugin","md":"https://skillfed.io/packages/jax-cuda13-plugin.md","pypi":"https://pypi.org/project/jax-cuda13-plugin/"},"maintenance":{"status":"active"},"meta":{"latest_release":"2026-07-16","license_spdx":null,"license_treatment":"permissive","name":"jax-cuda13-plugin","python_support":"supports_current","summary":"JAX Plugin for NVIDIA GPUs"},"popularity":{"monthly_downloads":420823,"position":6787,"tier":"top_15000"},"security":{"n_vulnerabilities":0},"version":"0.11.0"}