Before
After
Context for the issue:
Periodic HSGPs (now in PyMC 🥳 ) require iv(J, a) / exp(a) as normalizing factor. Using this directly could lead to overflows however (as iv grows like exp(x)). Instead ive(nu,x) = iv(nu,x) / exp(abs(x)) exists, and as a bonus cheaper to compute than the combined!
After #403 we already have an implementation of ive in JAX, and it's in scipy, so should be straightforward!