blob: 5344add1e739a7a9f432f9438245188746e9100d (
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
|
pkgbase = python-flax
pkgdesc = A neural network library and ecosystem for JAX designed for flexibility
pkgver = 0.10.5
pkgrel = 1
url = https://github.com/google/flax
arch = any
groups = jax
license = Apache-2.0
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-treescope
depends = python-typing_extensions
depends = python-yaml
optdepends = python-matplotlib: Export to TensorBoard.
optdepends = tensorboard: TensorBoard visualization and logging.
source = flax-0.10.5.tar.gz::https://github.com/google/flax/archive/refs/tags/v0.10.5.tar.gz
source = python-flax.diff
sha256sums = 0f5376597c31eed5033b17422fcb8314a90209d95958a2fba2a06e29c5eeb64f
sha256sums = SKIP
pkgname = python-flax
|