Package Usage: pypi: jax-cuda12-plugin
JAX Plugin for NVIDIA GPUs
14 versions
Latest release: about 1 year 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