license: apache-2.0
MambaBit. Bit-level cursed model with vocab size=2
4 layers, vocab size=2, embedded size = 4096 float32 parm per bit.
Training was done on first 8030848 bits of tiny Shakespeare in 10 hours on laptop with 16GB VRAM on 9 batches of 128*8 bit each. Training code included in trainer.ipynb
To run the model run
python mambabit.py "As sun raised over". Expected output
As sun raised over me.
LEONTES:
Now means means me not so much as my father,
In the good many lord, and my father come.
KING RICHARD III:
What is my father come and my father,
In the good lord, and my father come and before his father.
GLOUCESTER:
Now the goes of men, a
Bytes are encoded with most significant bit fed first. Eg '7' = [0, 0, 1, 1, 0, 1, 1, 1], so MSB 0 is being fed first rather than last as if it was with [1, 1, 1, 0, 1, 1, 0, 0]. Intuition with that is that bits at the beginning change less frequent than in the end, so model will be like "I think I will produce a digit" then "I think I will produce 7" instead of "so I spat something. Should it be a number? a letter? dunno"
I tried to use BF16 originally, but model went into nan (with default big LR) or gradients were so small weights didn't change(smaller LR). I switched back to F32, however some layers still initialize weight with factor x0.001 as I hoped it would stop model from going to nan.
Also next morning I realized that Mamba module doesn't come with built-in normalization and residual as I thought, which explains why BF16 attempts failed. Oops.