hypersphere
Quick implementation of nGPT, learning entirely on the hypersphere, from NvidiaAI
Vilenkin–Kuznetsov–Smorodinsky polyspherical (hyperspherical) coordinates in NumPy / PyTorch