Skip from the input to every layer: add the token embedding back after each block, x = block(x) + X0_MIX * x0. val_loss 1.539199 vs parent 1.560530 (2000 steps, seed 1337, eval as in train.py).
