pyproject.toml

[build-system]
requires = ["pdm-backend"]
build-backend = "pdm.backend"

[project]
name = "tiny-llm"
version = "0.1.0"
requires-python = ">=3.10, <3.13"
readme = "README.md"
dependencies = [
    "mlx>=0.25.0",
    "torch>=2.6.0",
    "torchtune>=0.6.1",
    "torchao>=0.10.0",
    "mlx-lm>=0.23.0",
    "numpy>=2.2.4",
    "pytest>=8.3.5",
    "ruff>=0.11.6",
    # this should not usually appear in a project dependency list but we add it to simplify the setup process
    "setuptools>=62",
    "nanobind==2.4.0"
]

[tool.pdm.scripts]
build-ext-ref.cmd = "python build.py"
build-ext-ref.working_dir = "src/extensions_ref"
main.cmd = "python main.py"
test.cmd = "pytest tests"
test-week1-ref.cmd = "pytest tests_ref_impl_week1"
test-week2-ref.cmd = "pytest tests_ref_impl_week2"
format = "ruff format"

[tool.pytest.ini_options]
addopts = [
    "--import-mode=importlib",
]
pythonpath = "src"