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