pyproject.toml

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

[project]
name = "liger_kernel"
version = "0.4.0"
description = "Efficient Triton kernels for LLM Training"
urls = { "Homepage" = "https://github.com/linkedin/Liger-Kernel" }
readme = { file = "README.md", content-type = "text/markdown" }
license = { file = "LICENSE" }
dependencies = [
    "torch>=2.1.2",
    "triton>=2.3.1",
]

[project.optional-dependencies]
transformers = [
    "transformers~=4.0"
]

dev = [
    "transformers>=4.44.2",
    "matplotlib>=3.7.2",
    "flake8>=4.0.1.1",
    "black>=24.4.2",
    "isort>=5.13.2",
    "pytest>=7.1.2",
    "datasets>=2.19.2",
    "torchvision>=0.16.2",
    "seaborn",
]

[tool.setuptools.packages.find]
where = ["src"]
include = ["liger_kernel", "liger_kernel.*"]

[tool.pytest.ini_options]
pythonpath = [
  "src",
  "."
]
asyncio_mode = "auto"
log_cli = true
log_cli_level = "INFO"