KeuralViT Stage 1 training in progress. Checkpoints every 4000 steps.