blob: d3d7144a0b49fea0008a994f7e04668fa67ac71e (
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
|
pkgbase = python-jaxlib-cuda
pkgdesc = XLA library for JAX
pkgver = 0.5.3
pkgrel = 1
url = https://github.com/jax-ml/jax/
arch = x86_64
groups = jax
license = Apache-2.0
makedepends = clang18
makedepends = python-build
makedepends = python-installer
makedepends = python-setuptools
makedepends = python-wheel
depends = cuda
depends = cudnn
depends = nccl
depends = python-absl
depends = python-ml-dtypes>=0.4.0
depends = python-numpy
depends = python-scipy
provides = python-jaxlib=0.5.3
conflicts = python-jaxlib
source = jaxlib-0.5.3.tar.gz::https://github.com/jax-ml/jax//archive/refs/tags/jax-v0.5.3.tar.gz
source = bazel-7.4.1-linux-x86_64::https://github.com/bazelbuild/bazel/releases/download/7.4.1/bazel-7.4.1-linux-x86_64
sha256sums = 1094581a30ec069965f4e3e67d60262570cc3dd016adc62073bc24347b14270c
sha256sums = c97f02133adce63f0c28678ac1f21d65fa8255c80429b588aeeba8a1fac6202b
pkgname = python-jaxlib-cuda
|