Package Usage: pypi: jax-cuda12-pjrt
JAX XLA PJRT Plugin for NVIDIA GPUs
14 versions
Latest release: about 1 year ago
1 dependent package
214,736 downloads last month
View more package details: https://packages.ecosystem.code.gouv.fr/registries/pypi.org/packages/jax-cuda12-pjrt