blob: d729d62d3e8881226422998335f430e46bd03c48 (
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.4
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.4.tar.gz::https://github.com/google-deepmind/optax/archive/refs/tags/v0.2.4.tar.gz
sha256sums = 5f1d94664b61d93841a0a36465d63e0ad939db9e638f8cbeb15d3a452c2b7542
pkgname = python-optax
|