pyproject.toml

[build-system]
requires = ["setuptools>=65.0"]
build-backend = "setuptools.build_meta"

[tool.setuptools.packages.find]
where = ["."]

[project]
name = "nanugpt"
version = "0.4.0"
authors = [
    { name = "Shital Shah", email = "sytelus@gmail.com" }
]
maintainers = [
    { name = "Shital Shah", email = "sytelus@gmail.com" }
]
description = "Simple, reliable and well tested training code for quick experiments with transformer based models"
readme = "README.md"
license = { text = "MIT" }
classifiers = [
    "Programming Language :: Python :: 3",
    "License :: OSI Approved :: MIT License",
    "Operating System :: OS Independent",
    "Intended Audience :: Science/Research"
]
keywords = ["gpt", "nanogpt", "tinystories", "grokking", "openwebtext", "pretraining"]
dependencies = [
    "einops",
    "tiktoken",
    "wandb",
    "mlflow",
    "sentencepiece",
    "tokenizers",
    "tiktoken",
    "transformers",
    "datasets",
    "tqdm",
    "matplotlib",
    "rich"
]

[project.urls]
Homepage = "https://github.com/sytelus/nanugpt"
Repository = "https://github.com/sytelus/nanugpt"