blob: f3a093d8b7f8245685db5b7edb010b744f249285 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
pkgbase = python-jaxlib-cuda
pkgdesc = XLA library for JAX
pkgver = 0.9.0
pkgrel = 3
url = https://github.com/jax-ml/jax
arch = x86_64
groups = jax
license = Apache-2.0
makedepends = clang20
makedepends = python-build
makedepends = python-installer
makedepends = python-setuptools
makedepends = python-wheel
depends = cuda
depends = cudnn
depends = nccl
depends = python-jaxlib
depends = python-ml-dtypes
depends = python-numpy
depends = python-scipy
noextract = bazel-7.7.0-linux-x86_64
source = jax-0.9.0.tar.gz::https://github.com/jax-ml/jax/archive/refs/tags/jax-v0.9.0.tar.gz
source = jax-xla-0.9.0.tar.gz::https://github.com/openxla/xla/archive/bb760b047bdbfeff962f0366ad5cc782c98657e0.tar.gz
source = bazel-7.7.0-linux-x86_64::https://github.com/bazelbuild/bazel/releases/download/7.7.0/bazel-7.7.0-linux-x86_64
source = bazelrc.sh
source = jax.diff
source = xla.diff
sha256sums = 8525c72ac7ea01851297df5b25ca4622c65299c265c87dfe78420bb29e7b1bb3
sha256sums = 665ec74f3ca69905ac21dbf075ccaeed97755d8e125e068a4297b5c6c35d3a5c
sha256sums = fe7e799cbc9140f986b063e06800a3d4c790525075c877d00a7112669824acbf
sha256sums = SKIP
sha256sums = SKIP
sha256sums = SKIP
pkgname = python-jaxlib-cuda
|