{"id":27881,"library":"jax-cuda13-plugin","title":"JAX CUDA 13 Plugin","description":"JAX plugin providing NVIDIA GPU support for CUDA 13.x. Version 0.10.0 is compatible with JAX v0.10.0. Release cadence follows JAX releases.","status":"active","version":"0.10.0","language":"python","source_language":"en","source_url":"https://github.com/jax-ml/jax","tags":["jax","cuda","gpu","plugin"],"install":[{"cmd":"pip install jax-cuda13-plugin","lang":"bash","label":"Install from PyPI"},{"cmd":"pip install jax[cuda13]==0.10.0 jax-cuda13-plugin==0.10.0","lang":"bash","label":"Install with JAX pin"}],"dependencies":[{"reason":"Required; plugin provides CUDA support for JAX.","package":"jax","optional":false},{"reason":"Required; JAX binary library.","package":"jaxlib","optional":false}],"imports":[{"note":"Plugin is imported as a side effect; no top-level API.","wrong":"","symbol":"jax_cuda13_plugin","correct":"import jax_cuda13_plugin"}],"quickstart":{"code":"import jax\nimport jax.numpy as jnp\nx = jnp.array([1., 2., 3.])\nprint(jax.devices())","lang":"python","description":"Verify GPU device is detected after installing plugin."},"warnings":[{"fix":"Install jax-cuda13-plugin==0.10.0 alongside jax==0.10.0.","message":"Plugin version must exactly match JAX major.minor version (e.g., 0.10.x). Mismatch causes import errors.","severity":"breaking","affected_versions":"all"},{"fix":"Verify CUDA version with `nvidia-smi`; install appropriate plugin (e.g., jax-cuda12-plugin for CUDA 12).","message":"jax-cuda13-plugin only supports CUDA 13.x; it will not work with other CUDA versions. Ensure your system has CUDA 13 runtime.","severity":"gotcha","affected_versions":"all"},{"fix":"Do not import plugin explicitly; just install and import jax.","message":"Importing jax_cuda13_plugin is not required; merely installing it enables GPU support. The import statement is a no-op and may cause confusion.","severity":"gotcha","affected_versions":"all"}],"env_vars":null,"last_verified":"2026-05-09T00:00:00.000Z","next_check":"2026-08-07T00:00:00.000Z","problems":[{"fix":"Install CUDA 13.x from NVIDIA and ensure LD_LIBRARY_PATH includes /usr/local/cuda-13/lib64.","cause":"CUDA runtime library not found; plugin requires CUDA 13.x installed.","error":"ImportError: libcuda.so.1: cannot open shared object file: No such file or directory"},{"fix":"Run `pip install jax-cuda13-plugin==0.10.0` matching your JAX version.","cause":"Plugin package not installed or version mismatch with JAX.","error":"jax._src.plugins.PluginNotInstalledError: jax_cuda13_plugin not installed"},{"fix":"Update NVIDIA driver to version that supports CUDA 13 (e.g., driver >= 545).","cause":"NVIDIA driver is too old for CUDA 13.","error":"RuntimeError: Failed to initialize XLA: CUDA driver version is insufficient for CUDA runtime version"}],"ecosystem":"pypi","meta_description":null,"install_score":null,"install_tag":null,"quickstart_score":null,"quickstart_tag":null}