blob: a9246162664f9d5f3c2804d8c7da178563f0a81c (
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
|
pkgbase = python-flax
pkgdesc = A neural network library and ecosystem for JAX designed for flexibility
pkgver = 0.10.1
pkgrel = 1
url = https://github.com/google/flax
arch = any
groups = jax
license = Apache
makedepends = python-build
makedepends = python-installer
makedepends = python-setuptools
makedepends = python-setuptools-scm
makedepends = python-wheel
depends = python-jax
depends = python-msgpack
depends = python-numpy
depends = python-optax
depends = python-orbax-checkpoint
depends = python-rich
depends = python-tensorstore
depends = python-typing_extensions
depends = python-yaml
optdepends = python-matplotlib: Export to TensorBoard.
optdepends = tensorboard: TensorBoard visualization and logging.
source = flax-0.10.1.tar.gz::https://github.com/google/flax/archive/refs/tags/v0.10.1.tar.gz
source = python-flax.diff
sha256sums = 806160b971a0eb940e362b928497b385035c70399048e339d59db8c3aa899d52
sha256sums = SKIP
pkgname = python-flax
|