SolveDelta/pyproject.toml at main · Yang916-yy/SolveDelta · 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
[build-system]
requires = ["setuptools>=68", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "causallsso"
version = "0.1.0"
description = "CUDA-native residual-frame Delta memory for causal language models"
readme = "README.md"
requires-python = ">=3.10"
keywords = [
    "causal-language-modeling",
    "cuda",
    "delta-rule",
    "flash-linear-attention",
    "linear-attention",
    "pytorch",
    "recurrent-memory",
    "triton",
]
classifiers = [
    "Development Status :: 3 - Alpha",
    "Environment :: GPU :: NVIDIA CUDA",
    "Intended Audience :: Science/Research",
    "Operating System :: POSIX :: Linux",
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3 :: Only",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
]
dependencies = [
    "torch>=2.7,<3",
    "transformers>=4.53.3,<6",
    # Production imports private FLA kernel modules. Keep this exact until a
    # newer release passes the full oracle, VJP, model, and graph suite.
    "flash-linear-attention==0.6.0",
]

[project.optional-dependencies]
native = [
    "causal-conv1d>=1.7.0,<2",
]
test = ["pytest>=8"]

[tool.setuptools.packages.find]
include = ["causallsso*"]

[project.urls]
Documentation = "https://github.com/Yang916-yy/SolveDelta/tree/main/docs"
Issues = "https://github.com/Yang916-yy/SolveDelta/issues"
Repository = "https://github.com/Yang916-yy/SolveDelta"

[tool.pytest.ini_options]
testpaths = ["tests"]