blob: c8f3db70cacc524266d54b2d05aab7ddb387e8c7 (
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
|
pkgbase = python-jaxlib-cuda
pkgdesc = XLA library for JAX
pkgver = 0.4.27
pkgrel = 1
url = https://github.com/google/jax/
arch = x86_64
groups = jax
license = Apache
makedepends = gcc12
makedepends = pybind11
makedepends = python-build
makedepends = python-installer
makedepends = python-setuptools
makedepends = python-wheel
depends = cuda
depends = cudnn
depends = nccl
depends = openssl
depends = python-absl
depends = python-flatbuffers
depends = python-ml-dtypes
depends = python-numpy
depends = python-scipy
provides = python-jaxlib=0.4.27
conflicts = python-jaxlib
source = jaxlib-0.4.27.tar.gz::https://github.com/google/jax/archive/refs/tags/jaxlib-v0.4.27.tar.gz
source = bazelrc.user
source = https://github.com/bazelbuild/bazel/releases/download/6.1.2/bazel-6.1.2-linux-x86_64
sha256sums = c2c82cd9ad3b395d5cbc0affa26a2938e52677a69ca8f0b9ef9922a52cac4f0c
sha256sums = 07da4c3594dad382ee02748b860c629ffa083ba37ad22a892291bdc72efbac5e
sha256sums = e89747d63443e225b140d7d37ded952dacea73aaed896bca01ccd745827c6289
pkgname = python-jaxlib-cuda
|