FilesExpand file tree

main

/

pyproject.toml

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

47 lines (44 loc) · 990 Bytes

main

/

pyproject.toml

Copy path

Top

File metadata and controls

• Code
• Blame

47 lines (44 loc) · 990 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

34

35

36

37

38

39

40

41

42

43

44

45

46

47

[project]

name = "mamba_ssm"

description = "Mamba state-space model"

readme = "README.md"

authors = [

{ name = "Tri Dao", email = "tri@tridao.me" },

{ name = "Albert Gu", email = "agu@cs.cmu.edu" }

]

requires-python = ">= 3.9"

dynamic = ["version"]

license = { file = "LICENSE" } # Include a LICENSE file in your repo

keywords = ["cuda", "pytorch", "state-space model"]

classifiers = [

"Programming Language :: Python :: 3",

"License :: OSI Approved :: Apache Software License",

"Operating System :: Unix"

]

dependencies = [

"torch",

"triton",

"ninja",

"einops",

"transformers",

"packaging",

"setuptools>=61.0.0",

]

[project.urls]

Repository = "https://github.com/state-spaces/mamba"

[project.optional-dependencies]

causal-conv1d = [

"causal-conv1d>=1.2.0"

]

dev = [

"pytest"

]

[build-system]

requires = [

"setuptools>=61.0.0",

"wheel",

"torch",

"packaging",

"ninja",

]

build-backend = "setuptools.build_meta"