package

python3.12-jax-cuda12-pjrt

Description

JAX XLA PJRT Plugin for NVIDIA GPUs.

Metadata

Version

0.5.3

License

Status

Source

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

Homepage

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

Maintainers (1)

Tomoya Otabi

Platforms (1)

Linux

Show all

x86_64-linux