GPT model for the paper: https://arxiv.org/abs/2302.05981