BitNet-b158 noteで実装を進めているBitLinear, BitLinearb158のコードとなります。 BitNet&BitNet b158の実装 pip install mybitnetでインストール可能です。 example/01_bit_llama_pretraining.ipynbファイルには、bitlinearを用いた事前学習の例があります。