MyNixOS website logo
MyNixOS website logo
Flakes
Categories
Options
Packages
package

python3.13-jax-cuda12-pjrt

Description

JAX XLA PJRT Plugin for NVIDIA GPUs.

Metadata

Version

0.5.0

License

Apache-2.0

Status

BrokenNo
InsecureNo
UnfreeNo
UnsupportedNo

Source

pkgs/development/python-modules/jax-cuda12-pjrt/default.nix:87

Homepage

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

Maintainers (1)

Platforms (2)

    Linux
Show all
  • aarch64-linux
  • x86_64-linux