This is your work, valued
nanoChatGPT. A crude RLHF layer on top of nanoGPT with Gumbel-Softmax trick