pyproject.toml

[project]
name = "scalable_softmax"
version = "0.1.1"
description = "PyTorch implementation of Scalable-Softmax for attention mechanisms"
authors = [
    { name = "Greg DeVosNouri", email = "gdevos010@gamil.com" }
]
readme = "README.md"
requires-python = ">= 3.9"
license = { file = "LICENSE" }
keywords = ["pytorch", "deep-learning", "attention", "transformer"]

classifiers=[
    'Development Status :: 4 - Beta',
    'Intended Audience :: Developers',
    'Topic :: Scientific/Engineering :: Artificial Intelligence',
    'License :: OSI Approved :: MIT License',
    'Programming Language :: Python :: 3.9',
]

dependencies = [
    "torch>=1.8",
    "numpy>=1.23.5"
]

[project.urls]
Homepage = "https://pypi.org/project/scalable-softmax/"
Repository = "https://github.com/gdevos010/Scalable-Softmax"

[project.optional-dependencies]
dev = [
    "ruff",
    "build",
    "twine"
]