jax-cuda12-pjrt

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


Downloads last day: 33,854
Downloads last week: 275,059
Downloads last month: 1,125,305