├── .gitignore ├── LICENSE ├── README.md ├── bitnet158 ├── __init__.py ├── bitlinear.py ├── bitlinear158.py ├── bitlinear158_inference.py └── inject.py ├── ds_config.json ├── examples └── ds_train.py ├── requirements.txt ├── setup.py └── test ├── test_grad.py ├── test_inference.py └── test_inject.py /.gitignore: -------------------------------------------------------------------------------- 1 | *.egg-info/ 2 | __pycache__/ 3 | /*.sh 4 | saved_model/ -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Entropy-xcy/bitnet158/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Entropy-xcy/bitnet158/HEAD/README.md -------------------------------------------------------------------------------- /bitnet158/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Entropy-xcy/bitnet158/HEAD/bitnet158/__init__.py -------------------------------------------------------------------------------- /bitnet158/bitlinear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Entropy-xcy/bitnet158/HEAD/bitnet158/bitlinear.py -------------------------------------------------------------------------------- /bitnet158/bitlinear158.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Entropy-xcy/bitnet158/HEAD/bitnet158/bitlinear158.py -------------------------------------------------------------------------------- /bitnet158/bitlinear158_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Entropy-xcy/bitnet158/HEAD/bitnet158/bitlinear158_inference.py -------------------------------------------------------------------------------- /bitnet158/inject.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Entropy-xcy/bitnet158/HEAD/bitnet158/inject.py -------------------------------------------------------------------------------- /ds_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Entropy-xcy/bitnet158/HEAD/ds_config.json -------------------------------------------------------------------------------- /examples/ds_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Entropy-xcy/bitnet158/HEAD/examples/ds_train.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Entropy-xcy/bitnet158/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Entropy-xcy/bitnet158/HEAD/setup.py -------------------------------------------------------------------------------- /test/test_grad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Entropy-xcy/bitnet158/HEAD/test/test_grad.py -------------------------------------------------------------------------------- /test/test_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Entropy-xcy/bitnet158/HEAD/test/test_inference.py -------------------------------------------------------------------------------- /test/test_inject.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Entropy-xcy/bitnet158/HEAD/test/test_inject.py --------------------------------------------------------------------------------