{"library":"jax-cuda12-pjrt","type":"library","category":null,"description":"The jax-cuda12-pjrt package provides the JAX XLA PJRT backend for NVIDIA GPUs, specifically built with CUDA 12. It serves as the `jaxlib` implementation when GPU acceleration is desired. The current version is 0.9.2, and JAX along with its ecosystem components typically follow a rapid release cadence, often with monthly or bi-monthly updates.","language":"python","status":"active","version":"0.9.2","tags":["deep learning","machine learning","gpu","cuda","xla","numpy","auto-differentiation","scientific computing"],"last_verified":"Fri May 22","install":[{"cmd":"pip install jax jax-cuda12-pjrt","imports":["import jax"]}],"homepage":null,"github":"https://github.com/jax-ml/jax","docs":null,"changelog":null,"pypi":"https://pypi.org/project/jax-cuda12-pjrt/","npm":null,"openapi_spec":null,"status_page":null,"smithery":null,"compatibility":{"summary":{"python_range":"3.10–3.9","success_rate":50,"avg_install_s":15.4,"avg_import_s":1.85,"wheel_type":"wheel"},"url":"https://checklist.day/v1/registry/jax-cuda12-pjrt/compatibility"}}