GitHub repository: tcbegley/nanoChatGPT
Description: A crude RLHF layer on top of nanoGPT with Gumbel-Softmax trick