mase/pyproject.toml at main · DeepWok/mase · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
[project]
name = "mase"
version = "1.0.0"
description = "Machine-Learning Accelerator System Exploration Tools"
readme = "README.md"
requires-python = ">=3.11.9"

dependencies = [
    "torch>=2.6",
    "torchvision",
    "onnx",
    "black",
    "toml",
    "GitPython",
    "colorlog",
    "pytest",
    "pytorch-lightning",
    "transformers==4.57",
    "timm",
    "pytorch-nlp",
    "datasets==3.3.2",
    "evaluate==0.4.3",
    "ipython",
    "ipdb",
    "sentencepiece",
    "einops",
    "pybind11",
    "tabulate",
    "tensorboardx",
    "hyperopt @ git+https://github.com/hyperopt/hyperopt.git",
    "accelerate",
    "optuna",
    "stable-baselines3[extra]",
    "h5py",
    "scikit-learn",
    "scipy==1.14",
    "onnxruntime",
    "matplotlib",
    "sphinx-rtd-theme",
    "sphinx-needs>=4",
    "sphinx-test-reports @ git+https://github.com/useblocks/sphinx-test-reports",
    "sphinxcontrib-plantuml",
    "imageio",
    "imageio-ffmpeg",
    "opencv-python",
    "kornia",
    "ghp-import",
    "optimum[onnxruntime]>=1.25",
    "pytest-profiling",
    "myst_parser",
    "pytest-cov",
    "pytest-xdist",
    "pytest-sugar",
    "pytest-html",
    "lightning",
    "wandb",
    "bitarray",
    "bitstring>=4.2",
    "emoji",
    "numpy==2.2.4",
    "tensorboard",
    "onnxconverter-common",
    "absl-py",
    "sphinx-glpi-theme",
    "prettytable",
    "pyyaml",
    "pynvml",
    "cvxpy",
    "py-cpuinfo",
    "pandas",
    "psutil",
    "Pillow==10.4.0",
    "mpmath==1.3.0",
    "myst-nb",
    "sphinx-book-theme",
    "pydot",
    "attr-dot-dict",
    "ultralytics==8.3.235",
    # mase-triton is automatically installed on Linux x86_64
    # On other platforms, install manually if needed: pip install mase-triton
    "mase-triton>=0.0.6.post4; platform_machine == 'x86_64' and sys_platform == 'linux'",
]

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

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