e2a2 0.0.1__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (30) hide show
  1. e2a2-0.0.1/LICENSE +1 -0
  2. e2a2-0.0.1/PKG-INFO +25 -0
  3. e2a2-0.0.1/README.md +2 -0
  4. e2a2-0.0.1/pyproject.toml +36 -0
  5. e2a2-0.0.1/setup.cfg +4 -0
  6. e2a2-0.0.1/src/e2a2/__init__.py +17 -0
  7. e2a2-0.0.1/src/e2a2/actx_lib.py +972 -0
  8. e2a2-0.0.1/src/e2a2/core.py +1568 -0
  9. e2a2-0.0.1/src/e2a2/ctx_lib.py +665 -0
  10. e2a2-0.0.1/src/e2a2/e2a2_engine.c +372 -0
  11. e2a2-0.0.1/src/e2a2/e2a2_engine.h +135 -0
  12. e2a2-0.0.1/src/e2a2/engine.py +294 -0
  13. e2a2-0.0.1/src/e2a2/entropy.py +1029 -0
  14. e2a2-0.0.1/src/e2a2/entropy2.py +678 -0
  15. e2a2-0.0.1/src/e2a2/eval_val.py +578 -0
  16. e2a2-0.0.1/src/e2a2/export.py +318 -0
  17. e2a2-0.0.1/src/e2a2/ggd_huffman.c +218 -0
  18. e2a2-0.0.1/src/e2a2/ggd_huffman.h +123 -0
  19. e2a2-0.0.1/src/e2a2/inference.py +112 -0
  20. e2a2-0.0.1/src/e2a2/law_coder.py +394 -0
  21. e2a2-0.0.1/src/e2a2/rl_coder.c +200 -0
  22. e2a2-0.0.1/src/e2a2/sweep.py +844 -0
  23. e2a2-0.0.1/src/e2a2/train.py +89 -0
  24. e2a2-0.0.1/src/e2a2/train_group.py +60 -0
  25. e2a2-0.0.1/src/e2a2/train_merged.py +106 -0
  26. e2a2-0.0.1/src/e2a2.egg-info/PKG-INFO +25 -0
  27. e2a2-0.0.1/src/e2a2.egg-info/SOURCES.txt +28 -0
  28. e2a2-0.0.1/src/e2a2.egg-info/dependency_links.txt +1 -0
  29. e2a2-0.0.1/src/e2a2.egg-info/requires.txt +8 -0
  30. e2a2-0.0.1/src/e2a2.egg-info/top_level.txt +1 -0
e2a2-0.0.1/LICENSE ADDED
@@ -0,0 +1 @@
1
+ Contact Dan Jacobellis
e2a2-0.0.1/PKG-INFO ADDED
@@ -0,0 +1,25 @@
1
+ Metadata-Version: 2.4
2
+ Name: e2a2
3
+ Version: 0.0.1
4
+ Summary: E2A2: Encoding Efficient Asymmetric Autoencoders
5
+ Author-email: Dan Jacobellis <danjacobellis@utexas.edu>
6
+ License-Expression: LicenseRef-Proprietary
7
+ Project-URL: Repository, https://github.com/danjacobellis/E2A2
8
+ Project-URL: Issues, https://github.com/danjacobellis/E2A2/issues
9
+ Classifier: Programming Language :: Python :: 3
10
+ Classifier: Operating System :: OS Independent
11
+ Requires-Python: >=3.10
12
+ Description-Content-Type: text/markdown
13
+ License-File: LICENSE
14
+ Requires-Dist: torch
15
+ Requires-Dist: torchvision
16
+ Requires-Dist: timm
17
+ Requires-Dist: einops
18
+ Requires-Dist: gigatorch
19
+ Requires-Dist: numpy
20
+ Requires-Dist: scipy
21
+ Requires-Dist: datasets
22
+ Dynamic: license-file
23
+
24
+ # E2A2
25
+ Encoding Efficient Asymmetric Autoencoders
e2a2-0.0.1/README.md ADDED
@@ -0,0 +1,2 @@
1
+ # E2A2
2
+ Encoding Efficient Asymmetric Autoencoders
e2a2-0.0.1/pyproject.toml ADDED
@@ -0,0 +1,36 @@
1
+ [build-system]
2
+ requires = ["setuptools>=42", "wheel"]
3
+ build-backend = "setuptools.build_meta"
4
+
5
+ [project]
6
+ name = "e2a2"
7
+ version = "0.0.1"
8
+ description = "E2A2: Encoding Efficient Asymmetric Autoencoders"
9
+ authors = [
10
+ {name = "Dan Jacobellis", email = "danjacobellis@utexas.edu"}
11
+ ]
12
+ readme = "README.md"
13
+ license = "LicenseRef-Proprietary"
14
+ license-files = ["LICENSE"]
15
+ classifiers = [
16
+ "Programming Language :: Python :: 3",
17
+ "Operating System :: OS Independent"
18
+ ]
19
+ dependencies = [
20
+ "torch",
21
+ "torchvision",
22
+ "timm",
23
+ "einops",
24
+ "gigatorch",
25
+ "numpy",
26
+ "scipy",
27
+ "datasets",
28
+ ]
29
+ requires-python = ">=3.10"
30
+
31
+ [tool.setuptools.package-data]
32
+ "e2a2" = ["*.c", "*.h"]
33
+
34
+ [project.urls]
35
+ Repository = "https://github.com/danjacobellis/E2A2"
36
+ Issues = "https://github.com/danjacobellis/E2A2/issues"
e2a2-0.0.1/setup.cfg ADDED
@@ -0,0 +1,4 @@
1
+ [egg_info]
2
+ tag_build =
3
+ tag_date = 0
4
+
e2a2-0.0.1/src/e2a2/__init__.py ADDED
@@ -0,0 +1,17 @@
1
+ """E2A2: the FRAPPEv3 split trainer, packaged (recipe unchanged).
2
+
3
+ Entry points (each importable via run(argv=None, **kwargs) and runnable
4
+ as a CLI via `python3 -m e2a2.<name>`):
5
+
6
+ - `train_group` -- GROUP phase: one group per invocation; real
7
+ entropy-coded rate known (and round-trip verified) before any merged
8
+ compute.
9
+ - `train_merged` -- MERGED phase off a group-stage checkpoint; also
10
+ --verify_only.
11
+ - `train` -- sequential wrapper (group then merged, per group).
12
+ - `inference` -- thin, PROVISIONAL checkpoint round-trip helpers (the
13
+ checkpoint format is expected to change).
14
+
15
+ Heavy imports (torch, datasets, the C-backed coder) happen when a
16
+ submodule is imported, not here.
17
+ """