Project URL: https://github.com/zion645/nano_gtp/blob/master/nano_gpt.ipynb