Quick implementation of nGPT, learning entirely on the hypersphere, from NvidiaAI - View it on GitHub