Package Usage: pypi: jax-cuda12-pjrt
JAX XLA PJRT Plugin for NVIDIA GPUs
14 versions
Latest release: plus d'un an 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