summaryrefslogtreecommitdiff
path: root/pyproject.toml
blob: 41c97038e4fedca0cdd8aa6ff564e562e9e9a93e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
[build-system]
requires = ["setuptools>=68"]
build-backend = "setuptools.build_meta"

[project]
name = "zbp-scaling"
version = "0.1.0"
description = "Scaling study for zeroth-order backpropagation (ZBP) vs BP on decoder-only language models"
readme = "README.md"
requires-python = ">=3.10"
license = { text = "Apache-2.0" }
dependencies = ["torch>=2.1", "numpy", "pyyaml", "tiktoken", "datasets"]

[project.optional-dependencies]
dev = ["pytest"]

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