File size: 1,023 Bytes
9639af0 | 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 | [build-system]
requires = ["setuptools>=68.0"]
build-backend = "setuptools.build_meta"
[project]
name = "laughlm"
version = "0.1.0"
description = "Configuration-driven decoder-only LLM training system built with JAX"
readme = "README.md"
requires-python = ">=3.9"
authors = [
{ name = "LaughLM Team" }
]
license = { text = "MIT" }
dependencies = [
"jax>=0.4.20",
"flax>=0.7.0",
"optax>=0.1.7",
"PyYAML>=6.0",
"numpy>=1.24",
"tqdm>=4.65",
]
[project.optional-dependencies]
dev = [
"black>=23.0",
"ruff>=0.1.0",
"mypy>=1.0",
"pytest>=7.0",
]
[tool.setuptools]
include-package-data = true
[tool.setuptools.packages.find]
where = ["."]
include = [
"config*",
"model*",
"training*",
"data*",
"utils*",
"distributed*",
"evaluation*",
]
[tool.black]
line-length = 88
target-version = ["py39"]
[tool.ruff]
line-length = 88
target-version = "py39"
select = ["E", "F", "W"]
[tool.mypy]
python_version = "3.9"
strict = true
ignore_missing_imports = true
|