I'm aware that JIT requires statically shaped arrays, which prevents using np.arange in some cases. However, I seem to have stumbled across a case that does have static shapes and still raises an ...