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