1  [build-system]
 2  requires = ["setuptools>=42", "wheel"]
 3  build-backend = "setuptools.build_meta"
 4  
 5  [project]
 6  name = "liger_kernel"
 7  version = "0.4.2"
 8  description = "Efficient Triton kernels for LLM Training"
 9  urls = { "Homepage" = "https://github.com/linkedin/Liger-Kernel" }
10  readme = { file = "README.md", content-type = "text/markdown" }
11  license = { file = "LICENSE" }
12  dependencies = [
13      "torch>=2.1.2",
14      "triton>=2.3.1",
15  ]
16  
17  [project.optional-dependencies]
18  transformers = [
19      "transformers~=4.0"
20  ]
21  
22  dev = [
23      "transformers>=4.44.2",
24      "matplotlib>=3.7.2",
25      "flake8>=4.0.1.1",
26      "black>=24.4.2",
27      "isort>=5.13.2",
28      "pytest>=7.1.2",
29      "datasets>=2.19.2",
30      "torchvision>=0.16.2",
31      "seaborn",
32  ]
33  
34  [tool.setuptools.packages.find]
35  where = ["src"]
36  include = ["liger_kernel", "liger_kernel.*"]
37  
38  [tool.pytest.ini_options]
39  pythonpath = [
40    "src",
41    "."
42  ]
43  asyncio_mode = "auto"
44  log_cli = true
45  log_cli_level = "INFO"