blob: 42933bf61ac493c6f90eb839732058d107f850bd (
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
|
pkgbase = python-mamba-ssm
pkgdesc = Mamba state-space model: selective-scan SSM layers with CUDA kernels
pkgver = 2.3.2.post1
pkgrel = 1
url = https://github.com/state-spaces/mamba
arch = x86_64
license = Apache-2.0
makedepends = python-build
makedepends = python-installer
makedepends = python-wheel
makedepends = python-setuptools
makedepends = python-packaging
makedepends = cuda
makedepends = gcc15
makedepends = ninja
depends = python
depends = python-pytorch
depends = python-causal-conv1d
depends = python-triton
depends = python-einops
depends = python-huggingface-hub
depends = python-transformers
optdepends = python-tilelang: Mamba-3 fused tilelang kernels
optdepends = python-apache-tvm-ffi: Mamba-3 CuTe step-function (tvm-ffi backend)
optdepends = python-cuda: Mamba-3 CuTe step-function (CUDA driver bindings)
source = https://files.pythonhosted.org/packages/source/m/mamba-ssm/mamba_ssm-2.3.2.post1.tar.gz
sha256sums = 104cc47e9101e5401a675fa2b784f2952b9b037f3b1dd83b5ac544394e95d028
pkgname = python-mamba-ssm
|