File size: 2,330 Bytes
e2bfccc | 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 91 92 93 | [build-system]
requires = ["setuptools>=61.0", "wheel"]
build-backend = "setuptools.build_meta"
[project]
name = "gamma-ssm-s4-enhanced"
version = "0.1.0"
description = "Gamma-structured SSM blocks with S4-inspired stability and full-sequence paths"
readme = "README.md"
requires-python = ">=3.8"
license = {text = "MIT"}
authors = [
{name = "StarMists"}
]
keywords = ["ssm", "state-space-models", "s4", "hippo", "mamba", "sequence-modeling", "gamma-ssm"]
classifiers = [
"Development Status :: 3 - Alpha",
"Intended Audience :: Developers",
"Intended Audience :: Science/Research",
"License :: OSI Approved :: MIT License",
"Operating System :: OS Independent",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.8",
"Programming Language :: Python :: 3.9",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
]
dependencies = [
"torch>=1.12.0",
"numpy>=1.20.0",
]
# Optional performance optimizations
[project.optional-dependencies]
dev = [
"pytest>=7.0",
"pytest-cov",
"build",
"wheel",
"black",
"isort",
"flake8",
"mypy",
]
notebook = [
"matplotlib>=3.6",
"pandas>=1.5",
"seaborn>=0.12",
"jupyter>=1.0",
"torchvision>=0.13",
]
performance = [
"triton>=2.0.0;platform_system!='Windows'",
]
[project.urls]
Homepage = "https://github.com/StarMists/gamma_SSM_S4_enhanced"
Documentation = "https://github.com/StarMists/gamma_SSM_S4_enhanced#readme"
Repository = "https://github.com/StarMists/gamma_SSM_S4_enhanced.git"
Issues = "https://github.com/StarMists/gamma_SSM_S4_enhanced/issues"
[tool.setuptools]
include-package-data = true
[tool.setuptools.packages.find]
include = ["gamma_space_model*", "csrc*"]
exclude = [
"benchmarks*",
"examples*",
"output*",
"scripts*",
"tests*",
]
[tool.pytest.ini_options]
testpaths = ["tests"]
[tool.black]
line-length = 100
target-version = ['py38', 'py39', 'py310', 'py311']
[tool.isort]
profile = "black"
line_length = 100
[tool.mypy]
python_version = "3.8"
warn_return_any = true
warn_unused_configs = true
disallow_untyped_defs = false
|