Package Usage: pypi: jax-cuda12-plugin
JAX Plugin for NVIDIA GPUs
14 versions
Latest release: environ un an ago
1 dependent package
220 033 downloads last month
View more package details: https://packages.ecosystem.code.gouv.fr/registries/pypi.org/packages/jax-cuda12-plugin