{"id":27880,"library":"jax-cuda13-pjrt","title":"JAX CUDA 13 PJRT Plugin","description":"JAX PJRT plugin for NVIDIA GPUs using CUDA 13. Provides the XLA compiler backend for JAX on NVIDIA hardware. Version 0.10.0 aligns 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","cuda13","pjrt","nvidia","gpu","xla"],"install":[{"cmd":"pip install jax-cuda13-pjrt","lang":"bash","label":"Install from PyPI"}],"dependencies":[{"reason":"JAX core is required; the plugin is a backend.","package":"jax","optional":false},{"reason":"Required for XLA runtime; must match CUDA version.","package":"jaxlib","optional":false}],"imports":[{"wrong":"import jax_cuda13_pjrt","symbol":"jax_plugins","correct":"import jax_plugins"}],"quickstart":{"code":"import jax\nimport jax.numpy as jnp\nimport jax_cuda13_pjrt  # noqa: F401, ensure NVIDIA GPU plugin is registered\n\n# Verify device is visible\nprint(jax.devices())\n\n# Simple computation\nx = jnp.array([1, 2, 3])\ny = jnp.square(x)\nprint(y)","lang":"python","description":"Import the plugin to register the CUDA 13 backend, then use JAX as usual."},"warnings":[{"fix":"Always import jax_cuda13_pjrt at the top of your script.","message":"The plugin must be imported before any JAX computation to ensure the GPU backend is selected.","severity":"gotcha","affected_versions":"all"},{"fix":"Ensure jax==0.10.0 and jax-cuda13-pjrt==0.10.0 are installed together.","message":"Plugin version must match the JAX version. Using mismatched versions may cause runtime errors or undefined behavior.","severity":"breaking","affected_versions":"all"},{"fix":"Verify your CUDA version with nvcc --version. If CUDA < 13, use jax-cuda12-pjrt instead.","message":"The plugin is only for CUDA 13.x; it will not work with older CUDA toolkits.","severity":"gotcha","affected_versions":"all"}],"env_vars":null,"search_vec":"'0.10.0':26 '13':3,14 'align':27 'backend':19 'cadenc':32 'compil':18 'cuda':2,13 'cuda13':37 'follow':33 'gpu':40 'gpus':11 'hardwar':24 'jax':1,6,21,29,34,36 'nvidia':10,23,39 'pjrt':4,7,38 'plugin':5,8 'provid':15 'releas':31,35 'use':12 'v0.10.0':30 'version':25 'xla':17,41","created_at":"2026-05-09T05:52:34.459610+00:00","updated_at":"2026-05-09T05:52:34.459610+00:00","problems":null,"ecosystem":"pypi","meta_description":null,"install_score":null,"quickstart_score":null,"quickstart_tag":null,"pypi_latest":null,"cli_name":"","cli_version":null,"type":"library","homepage":null,"github":"https://github.com/jax-ml/jax","docs":null,"changelog":null,"pypi":"https://pypi.org/project/jax-cuda13-pjrt/","npm":null,"openapi_spec":null,"status_page":null,"smithery":null,"categories":["ai-ml"],"base_url":null,"auth_type":null,"provenance":{"verified_status":"passing","verified_at":"2026-07-03","last_verified":"2026-07-03","next_check":"2026-08-02","install_tag":null}}