bakeNN/pyproject.toml at main · scienthoon/bakeNN · 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
[build-system]
requires = ["setuptools>=77"]
build-backend = "setuptools.build_meta"

[project]
name = "bakenn"
dynamic = ["version"]
description = "Static INT8 AOT compiler for fixed-shape microcontroller models"
readme = "README.md"
requires-python = ">=3.10"
license = "Apache-2.0"
license-files = ["LICENSE"]
keywords = ["int8", "aot-compiler", "microcontroller", "pytorch", "embedded-ml"]
classifiers = [
  "Development Status :: 4 - Beta",
  "Operating System :: OS Independent",
  "Programming Language :: Python :: 3",
  "Programming Language :: Python :: 3.10",
  "Programming Language :: Python :: 3.11",
  "Programming Language :: Python :: 3.12",
  "Programming Language :: Python :: 3.13",
  "Typing :: Typed",
  "Topic :: Scientific/Engineering :: Artificial Intelligence",
  "Topic :: Software Development :: Compilers",
]
dependencies = ["numpy>=1.24,<3"]

[project.urls]
Homepage = "https://github.com/scienthoon/bakeNN"
Repository = "https://github.com/scienthoon/bakeNN.git"
Issues = "https://github.com/scienthoon/bakeNN/issues"
Changelog = "https://github.com/scienthoon/bakeNN/blob/main/CHANGELOG.md"

[project.optional-dependencies]
torch = ["torch>=2.9,<2.11"]
tflite = ["flatbuffers>=23,<26", "tflite>=2.18,<3"]
tflite-verify = ["ai-edge-litert>=2.2,<3"]
model-zoo = ["torch>=2.9,<2.11", "torchvision>=0.24,<0.26"]
test = ["pytest>=8"]

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

[tool.setuptools.package-data]
"bakenn" = ["py.typed"]
"bakenn.backend.cmsis_nn.vendor" = [
  "cmsis_nn/Include/*.h",
  "cmsis_nn/Source/ConvolutionFunctions/*.c",
  "cmsis_nn/Source/FullyConnectedFunctions/*.c",
  "cmsis_nn/Source/NNSupportFunctions/*.c",
  "cmsis_nn/Source/PoolingFunctions/*.c",
  "cmsis_nn/LICENSE.txt",
  "cmsis_nn/BAKENN_PROVENANCE.md",
  "cmsis_core/Include/*.h",
  "cmsis_core/LICENSE.txt",
]
"bakenn.backend.esp_nn.vendor" = [
  "esp_nn/include/*.h",
  "esp_nn/src/*/*.h",
  "esp_nn/src/*/*.c",
  "esp_nn/src/*/*.S",
  "esp_nn/LICENSE",
  "esp_nn/BAKENN_PROVENANCE.md",
]

[tool.setuptools.dynamic]
version = { attr = "bakenn._version.VERSION" }