-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathpyproject.toml
More file actions
80 lines (72 loc) · 2.16 KB
/
Copy pathpyproject.toml
File metadata and controls
80 lines (72 loc) · 2.16 KB
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
[build-system]
requires = ["setuptools>=68", "wheel"]
build-backend = "setuptools.build_meta"
[project]
name = "temporal-mesh-transformer"
version = "0.1.0"
description = "TemporalMesh Transformer — dynamic graph attention with temporal decay and adaptive depth routing"
requires-python = ">=3.10"
readme = "README.md"
license = {text = "MIT"}
authors = [
{name = "Vignesh S", email = "applemacbook6sep2004@gmail.com"},
]
keywords = [
"transformer", "deep-learning", "pytorch", "graph-neural-network",
"adaptive-compute", "sparse-attention", "temporal-decay", "early-exit",
"novel-architecture", "language-model", "nlp",
]
classifiers = [
"Development Status :: 3 - Alpha",
"Intended Audience :: Science/Research",
"License :: OSI Approved :: MIT License",
"Operating System :: OS Independent",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
"Programming Language :: Python :: 3.12",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
"Topic :: Software Development :: Libraries :: Python Modules",
]
dependencies = [
"torch>=2.2.0",
"einops>=0.7.0",
]
[project.optional-dependencies]
train = [
"transformers>=4.40.0",
"datasets>=2.19.0",
"tokenizers>=0.19.0",
"wandb>=0.17.0",
"tensorboard>=2.16.0",
"pandas>=2.2.0",
"tabulate>=0.9.0",
]
notebooks = [
"jupyter>=1.0.0",
"matplotlib>=3.8.0",
]
dev = [
"pytest>=8.0.0",
"pytest-cov>=5.0.0",
"ruff>=0.4.0",
"numpy>=1.26.0",
]
[project.urls]
Homepage = "https://vignesh2027.github.io/TemporalMesh-Transformer"
Repository = "https://github.com/vignesh2027/TemporalMesh-Transformer"
Documentation = "https://vignesh2027.github.io/TemporalMesh-Transformer"
"Bug Tracker" = "https://github.com/vignesh2027/TemporalMesh-Transformer/issues"
Paper = "https://doi.org/10.5281/zenodo.20287197"
[tool.black]
line-length = 100
target-version = ["py310", "py311"]
[tool.ruff]
line-length = 100
exclude = ["tmt/experiments/"]
[tool.ruff.lint]
select = ["E", "F"]
[tool.setuptools.packages.find]
include = ["tmt*"]
[tool.pytest.ini_options]
testpaths = ["tests"]