Hyperspherical coordinates in NumPy / PyTorch / JAX.
High performance computational platform in Python for the spectral Galerkin method