package

python3.13-jax-cuda12-plugin

Description

JAX Plugin for CUDA12

Metadata

Version

0.11.1

License

Status

Source

pkgs/development/python-modules/jax-cuda12-plugin/default.nix:96

Homepage

https://github.com/jax-ml/jax/tree/main/jax_plugins/cuda

Maintainers (1)

Tomoya Otabi

Platforms (2)

Linux

Show all

aarch64-linux
x86_64-linux