blob: b24d1be7076b2d1438b6a9ce843571587d4f0088 (
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
|
pkgbase = python-jax-bin
pkgdesc = Composable transformations of Python+NumPy programs
pkgver = 0.5.1
pkgrel = 1
url = https://github.com/jax-ml/jax/
arch = any
license = Apache
makedepends = python-build
makedepends = python-installer
makedepends = python-wheel
depends = python-absl
depends = python-jaxlib>=0.4.35
depends = python-ml-dtypes
depends = python-numpy
depends = python-opt_einsum
depends = python-scipy
optdepends = python-protobuf: Australis
optdepends = python-requests: Distributed computations
provides = python-jax
conflicts = python-jax
source = https://files.pythonhosted.org/packages/source/j/jax/jax-0.5.1.tar.gz
sha256sums = c098f74846ee718165bbfa83521ae10cd52cf50b47f043f8b33a6cfd3c20ddfd
pkgname = python-jax-bin
|