nanoGPT is focusing on Transformer Pretraining stagenanoGPT StructurenanoGPT prepare.pynanoGPT model.pynanoGPT train.pynanoGPT sample.py 3D Visualization