On day 44, I found that replacing the contribution of layer 0 with a table looked up by token ID in a small 6-layer model ...