Code: www.github.com/soon-yau/upgpt

Paper: https://arxiv.org/abs/2304.08870