package

python3.14-jax-cuda13-plugin

Description

JAX Plugin for CUDA13

Metadata

Version

0.12.0.1

License

Status

Source

pkgs/development/python-modules/jax-cuda13-plugin/default.nix:97

Homepage

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

Maintainers (6)

Platforms (2)

Linux

Show all

aarch64-linux
x86_64-linux