jax-cuda13-pjrt

PyPI page
Home page
Author: JAX team
License: Apache-2.0
Summary: JAX XLA PJRT Plugin for NVIDIA GPUs
Latest version: 0.12.0


Downloads last day: 12,385
Downloads last week: 101,285
Downloads last month: 300,046
7-day smoothing