numba-cuda
Compute Jacobi polynomial from order 0 to N efficiently in NumPy / PyTorch / JAX / array API