blob: 8c11ef04b002dbf0f68a78eb1e12a403269de0bb (
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
32
33
|
pkgbase = python-deepspeed
pkgdesc = DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
pkgver = 0.12.4
pkgrel = 1
url = https://github.com/microsoft/DeepSpeed
arch = x86_64
license = Apache-2.0
makedepends = python-build
makedepends = python-installer
makedepends = python-wheel
depends = python
depends = python-hjson
depends = ninja
depends = python-numpy
depends = python-packaging
depends = python-psutil
depends = python-py-cpuinfo
depends = python-pydantic
depends = python-pynvml
depends = python-pytorch
depends = python-tqdm
depends = python-cupy
depends = python-huggingface-hub
depends = tensorboard
depends = python-transformers
depends = python-mpi4py
depends = python-tabulate
depends = python-fairscale
depends = python-safetensors
source = https://github.com/microsoft/DeepSpeed/archive/refs/tags/v0.12.4.tar.gz
sha256sums = 736dbfd0ce637749899218286f13d9a932dfbbf692bdbb3ee10d988c0b945327
pkgname = python-deepspeed
|