blob: c4c5845c7ffe5b5d1f6d4723261bd2ced27da292 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
pkgbase = python-optax
pkgdesc = A gradient processing and optimization library for JAX
pkgver = 0.2.8
pkgrel = 1
url = https://github.com/google-deepmind/optax
arch = any
groups = jax
license = Apache-2.0
makedepends = python-build
makedepends = python-installer
makedepends = python-flit-core
makedepends = python-wheel
depends = python-absl
depends = python-jax
depends = python-numpy
optdepends = python-dm-haiku: Deep learning framework from DeepMind
optdepends = python-flax: Deep learning framework from Google
source = optax-0.2.8.tar.gz::https://github.com/google-deepmind/optax/archive/refs/tags/v0.2.8.tar.gz
source = optax.diff
sha256sums = c5864f15c7f9ddfa3f703d8a1f73d5a516db07d220fb0a434741d23f45da4559
sha256sums = SKIP
pkgname = python-optax
|