xlstm/pyproject.toml at main · NX-AI/xlstm · GitHub

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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
[build-system]
requires = [
  "scikit-build-core>=0.11",
  "pybind11>=3.0",
  "numpy>=2.0",
  "torch>=2.0",
]
build-backend = "scikit_build_core.build"

[project]
name = "xlstm"
version = "2.0.5"
authors = [
  { name="Maximilian Beck", email="beck@ml.jku.at" },
  { name="Korbinian Poeppel", email="poeppel@ml.jku.at" },
  { name="Andreas Auer", email="auer@ml.jku.at" },
]
description = "A novel LSTM variant with promising performance compared to Transformers or State Space Models."
readme = "README.md"
license = {file="LICENSE"}
requires-python = ">=3.10"
classifiers = [
    "Programming Language :: Python :: 3",
    "Operating System :: OS Independent",
]
keywords = ["LSTM", "Transformer", "Machine Learning", "Deep Learning", "State Space Models"]
dependencies = [
    "torch>=2.0",
    "einops",
    "numpy",
    "opt_einsum",
    "omegaconf",
    "transformers",
    "reportlab",
    "joypy",
    "ipykernel",
    "dacite",
    "ftfy",
    "ninja",
    "huggingface-hub",
    "rich",
    "tokenizers",
    "tqdm",
    "seaborn",
    "mlstm_kernels; python_version >= '3.11'",
]

[tool.uv]
cache-keys = [
    {file = "pyproject.toml"},
    {file = "CMakeLists.txt"},
    {file = "blocks/slstm/src/cuda/*.c"},
    {file = "blocks/slstm/src/cuda/*.cc"},
    {file = "blocks/slstm/src/cuda/*.h"},
    {file = "blocks/slstm/src/cuda/*.cu"},
    {file = "blocks/slstm/src/cuda/*.cuh"},
    {file = "blocks/slstm/src/util/*.c"},
    {file = "blocks/slstm/src/util/*.cc"},
    {file = "blocks/slstm/src/util/*.h"},
    {file = "blocks/slstm/src/util/*.cu"},
    {file = "blocks/slstm/src/util/*.cuh"},
]

[[tool.uv.index]]
name = "pytorch-cu126"
url = "https://download.pytorch.org/whl/cu126"
explicit = true

[[tool.uv.index]]
name = "pytorch-cu130"
url = "https://download.pytorch.org/whl/cu130"
explicit = true

[tool.uv.sources]
torch = [
  { index = "pytorch-cu126", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
  # { index = "pytorch-cu130", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
]


[tool.scikit-build]
minimum-version = "build-system.requires"


[tool.setuptools.package-data]
"xlstm" = ["blocks/slstm/src/cuda/*.c", "blocks/slstm/src/cuda/*.cc", "blocks/slstm/src/cuda/*.h", "blocks/slstm/src/cuda/*.cu", "blocks/slstm/src/cuda/*.cuh", "blocks/slstm/src/util/*.c", "blocks/slstm/src/util/*.cc", "blocks/slstm/src/util/*.h", "blocks/slstm/src/util/*.cu", "blocks/slstm/src/util/*.cuh"]

[project.urls]
Repository = "https://github.com/NX-AI/xlstm"
Issues = "https://github.com/NX-AI/xlstm/issues"