thunder/pyproject.toml at main · ecstayalive/thunder · 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
[build-system]
requires = ["setuptools>=61.0"]
build-backend = "setuptools.build_meta"

[project]
name = "thunder"
version = "0.0.0"
authors = [{ name = "ecstayalive", email = "ecstayalive@163.com" }]
description = ""
readme = "README.md"
requires-python = ">=3.11"
classifiers = ["Programming Language :: Python :: 3"]
dependencies = [
    "coolname",
    "gymnasium",
    "numpy",
    "rich",
    "rich_argparse",
    "tqdm",
]

[tool.setuptools]
include-package-data = true

[tool.setuptools.packages.find]
where = ["."]
include = ["thunder", "thunder.*"]

[project.urls]
"Homepage" = "https://github.com/ecstayalive/thunder"
"Bug Tracker" = "https://github.com/ecstayalive/thunder/issues"
"Discussions" = "https://github.com/ecstayalive/thunder/discussions"

[project.scripts]

[project.optional-dependencies]
# Official prebuilt wheels (no local nvcc): https://github.com/state-spaces/mamba/releases/tag/v2.3.1
# Match torch 2.7 + cu12 + cp311 + cxx11abiTRUE. Other platforms fall back to PyPI sdist (needs nvcc).
torch = [
    "torch<=2.10",
    "optree>=0.19.0",
    "mamba-ssm @ https://github.com/state-spaces/mamba/releases/download/v2.3.1/mamba_ssm-2.3.1%2Bcu12torch2.7cxx11abiTRUE-cp311-cp311-linux_x86_64.whl ; sys_platform == 'linux' and platform_machine == 'x86_64'",
    "mamba-ssm @ https://github.com/state-spaces/mamba/releases/download/v2.3.1/mamba_ssm-2.3.1%2Bcu12torch2.7cxx11abiTRUE-cp311-cp311-linux_aarch64.whl ; sys_platform == 'linux' and platform_machine == 'aarch64'",
    "mamba-ssm==2.3.1 ; sys_platform != 'linux' or (sys_platform == 'linux' and platform_machine != 'x86_64' and platform_machine != 'aarch64')",
]
jax = ["jax[cuda12]<=0.6", "flax<=0.11", "optax<=0.3"]
warp = ["warp-lang"]
all = ["thunder[torch,jax,warp]"]