.SRCINFO


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

pkgbase = python-jax-rocm
	pkgdesc = XLA library for JAX (jaxlib for ROCM)
	pkgver = 0.4.31
	pkgrel = 1
	url = https://github.com/google/jax
	arch = x86_64
	groups = jax
	license = Apache
	makedepends = python-build
	makedepends = python-installer
	makedepends = python-setuptools
	makedepends = python-wheel
	makedepends = bazel
	makedepends = miopen
	makedepends = rccl
	makedepends = rocm-hip-sdk
	depends = python-absl
	depends = python-flatbuffers
	depends = python-ml-dtypes
	depends = python-numpy
	depends = python-scipy
	depends = miopen
	depends = rccl
	depends = rocm-hip-runtime
	provides = python-jaxlib=0.4.31
	conflicts = python-jaxlib
	source = jax-jaxlib-v0.4.31.tar.gz::https://github.com/google/jax/archive/refs/tags/jaxlib-v0.4.31.tar.gz
	source = xla-rocm-jaxlib-v0.4.31.tar.gz::https://github.com/ROCmSoftwarePlatform/xla/archive/refs/heads/rocm-jaxlib-v0.4.31.tar.gz
	sha256sums = 022ea1347f9b21cbea31410b3d650d976ea4452a48ea7317a5f91c238031bf94
	sha256sums = 2ea3a38396a679cb6bd4e737c759f87357ba4bc20c776a1c61e90f09a4c52779

pkgname = python-jax-rocm