pyproject.toml

[project]
name = "reasoning-pretraining-alignment"
version = "0.1.0"
description = "Reasoning-aware pretraining reduces alignment cost study"
requires-python = ">=3.10"
dependencies = [
    "torch>=2.2.0",
    "transformers>=4.45.0",
    "accelerate>=0.27.0",
    "datasets>=2.18.0",
    "tokenizers>=0.19.0",
    "safetensors>=0.4.2",
    "sentencepiece>=0.2.0",
    "trl>=0.8.0",
    "peft>=0.9.0",
    "bitsandbytes>=0.43.0",
    "wandb>=0.16.3",
    "hydra-core>=1.3.2",
    "omegaconf>=2.3.0",
    "tensorboard>=2.16.2",
    "evaluate>=0.4.1",
    "scikit-learn>=1.4.0",
    "scipy>=1.12.0",
    "rouge-score>=0.1.2",
    "numpy>=1.24.3",
    "pandas>=2.2.0",
    "matplotlib>=3.8.2",
    "seaborn>=0.13.2",
    "tqdm>=4.66.1",
    "rich>=13.7.0",
    "typer>=0.9.0",
    "pydantic>=2.6.0",
    "einops>=0.7.0",
    "jupyter>=1.0.0",
    "ipython>=8.0.0",
]

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

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

[project.optional-dependencies]
dev = [
    "pytest>=7.0.0",
    "black>=23.0.0",
    "ruff>=0.1.0",
    "mypy>=1.0.0",
]

flash = [
    "flash-attn>=2.5.5",
    "triton>=2.2.0",
]