STAPLE/pyproject.toml at main · Infini-AI-Lab/STAPLE · 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
# ---- All project specifications ---- #
[project]
name = "staple"
description = "Static sparse upcycling with per-layer embedding memory, on a torchtitan base"
readme = "README.md"
requires-python = ">=3.10"
license = {file = "LICENSE"}
authors = [
    { name = "Ranajoy Sadhukhan" },
    { name = "Euijin Hong" },
    { name = "Sheng Cao" },
    { name = "Zechun Liu" },
    { name = "Shuowei Jin" },
    { name = "Sha Li" },
    { name = "Changlong Yu" },
    { name = "Xin Liu" },
    { name = "Qin Lu" },
    { name = "Zhenyu Shi" },
    { name = "Priyanka Nigam" },
    { name = "Beidi Chen" },
]
keywords = ["pytorch", "training", "llm", "sparse upcycling", "static sparsity"]
dependencies = [
    # Stateful Dataloader
    "torchdata>=0.8.0",

    # Hugging Face integrations
    "datasets>=3.6.0",
    "tokenizers>=0.15.0",
    "safetensors",

    # Miscellaneous
    "fsspec",
    "tyro",
    "tensorboard",
    "wandb",
    "einops",
    "pillow",
]
dynamic = ["version"]

[project.urls]
Homepage = "https://infini-ai-lab.github.io/STAPLE"
Paper = "https://arxiv.org/abs/2610.12599"
GitHub = "https://github.com/Infini-AI-Lab/STAPLE"
Documentation = "https://github.com/Infini-AI-Lab/STAPLE/tree/main/docs"
Issues = "https://github.com/Infini-AI-Lab/STAPLE/issues"

[project.optional-dependencies]
dev = [
    "pre-commit",
    "pytest",
    "pytest-cov",
    "expecttest", # test_tokenizer
    "pyrefly==0.45.1",
]

[tool.setuptools.dynamic]
version = {file = "assets/version.txt"}

[tool.codespell]
skip = ["*.json"]
ignore-words-list = ["assertin", "datas" ,"indx", "inpt", "nd", "socio-economic"]

# ---- Explicit project build information ---- #
[build-system]
requires = ["setuptools>=61.0"]
build-backend = "setuptools.build_meta"

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

[tool.pytest.ini_options]
addopts = ["--showlocals"]  # show local variables in tracebacks
testpaths = ["tests"]

[tool.pyrefly]
project-excludes = ["torchtitan/experiments", "**/tests/**"]
replace-imports-with-any = ["torchao.*", "torchft", "torchvision.*", "deep_ep.*", "jinja2.*"]  # optional dependencies
search-path = ["../pytorch"]  # local built pytorch