[build-system]
requires = ["setuptools>=68", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "layer-interactions"
version = "0.1.0"
description = "Second-order interaction pruning for Hugging Face causal language models"
readme = "README.md"
requires-python = ">=3.10"
authors = [
  {name = "Angshul Majumdar"}
]
dependencies = [
  "torch>=2.2",
  "transformers>=4.45",
  "accelerate>=1.0",
  "psutil>=5.9"
]

[project.optional-dependencies]
examples = ["datasets>=3.0"]
test = ["pytest>=8.0"]

[tool.setuptools.packages.find]
where = ["."]
include = ["layer_interactions*"]