blob: 0604ab7ff3a1d3a64229fc0b31801866d56716b0 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
pkgbase = python-optax
pkgdesc = A gradient processing and optimization library for JAX
pkgver = 0.2.3
pkgrel = 1
url = https://github.com/google-deepmind/optax
arch = any
groups = jax
license = Apache
makedepends = python-build
makedepends = python-installer
makedepends = python-flit-core
makedepends = python-wheel
depends = python-absl
depends = python-chex
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.3.tar.gz::https://github.com/google-deepmind/optax/archive/refs/tags/v0.2.3.tar.gz
sha256sums = 0a794c0756c565b2b03db0a133f9e8dae09ad306b239ab7fe225b4a85feb7f03
pkgname = python-optax
|