blob: df62aba18c755a846680bded8e23834ac82d65ee (
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
|
pkgbase = python-jaxlib-cuda
pkgdesc = XLA library for JAX
pkgver = 0.4.7
pkgrel = 2
url = https://github.com/google/jax/
arch = x86_64
groups = jax
license = Apache
makedepends = python-installer
makedepends = python-setuptools
makedepends = python-wheel
depends = absl-py
depends = cuda
depends = cudnn
depends = nccl
depends = python-flatbuffers
depends = python-ml-dtypes
depends = python-numpy
depends = python-scipy
provides = python-jaxlib
conflicts = python-jaxlib
source = jaxlib-0.4.7.tar.gz::https://github.com/google/jax/archive/refs/tags/jaxlib-v0.4.7.tar.gz
sha256sums = 0578d5dd5035b5225cadb6a62ca5f93dd76b70292268502fc01a0fd9ca7001d0
pkgname = python-jaxlib-cuda
|