pyproject.toml

[tool.poetry]
name = "TransformerX"
version = "0.0.1"
description = "TransformerX is a python library for building transformer-based models using pre-built layers."
homepage = "https://github.com/tensorops/TransformerX"
repository = "https://github.com/tensorops/TransformerX"
authors = ["Soran Ghaderi <soran.gdr.cs@gmail.com>", "sigma1326 <taleb.zarhesh@gmail.com>"]
maintainers = ["Soran Ghaderi <soran.gdr.cs@gmail.com>", "sigma1326 <taleb.zarhesh@gmail.com>"]
readme = "README.md"
license = "MIT"
include = [
    "License",
]
packages = [
    { include = "transformerx" }
]
classifiers = [
    "Development Status :: 4 - Beta",
    "Intended Audience :: Developers",
    "License :: OSI Approved :: BSD License",
    "Operating System :: OS Independent",
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.6",
    "Programming Language :: Python :: 3.7",
    "Programming Language :: Python :: 3.8",
    "Programming Language :: Python :: 3.9",
    "Programming Language :: Python :: 3.10",
    "Programming Language :: Python :: 3.11",
    "Topic :: Software Development",
    "Topic :: Software Development :: Libraries",
    "Topic :: Software Development :: Libraries :: Python Modules",
    "Topic :: Scientific/Engineering",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
    "Topic :: Scientific/Engineering :: Mathematics",
    "Intended Audience :: Science/Research",
    "Intended Audience :: Developers",
    "Intended Audience :: Education",
]
keywords = [
    "transformer",
    "deep-learning",
    "machine-learning",
    "NLP",
    "natural-language-processing",
    "computer-vision",
    "cv",
    "vision",
    "speech-recognition",
]

[tool.poetry.dependencies]
numpy = ">=1.18.5"
matplotlib = {version = "^3.5.3", optional = true }
tensorflow = ">=2.2.0"
einops = "^0.4.1"

[tool.poetry.dev-dependencies]
pytest = "^7.1.2"

[build-system]
requires = ["poetry-core>=1.0.0"]
build-backend = "poetry.core.masonry.api"