One of the challenges with JAX is that you often can't track down pesky NaN and edge cases. This is because sometimes a computation is very slow when run with disable_jit such that it takes too long to find the NaN the old fashioned was. And, creating a reproducible example with a smaller computation is often not reasonable. This issue is a catch-all for such non-terminating behaviour.
One of the challenges with JAX is that you often can't track down pesky NaN and edge cases. This is because sometimes a computation is very slow when run with
disable_jit
such that it takes too long to find the NaN the old fashioned was. And, creating a reproducible example with a smaller computation is often not reasonable. This issue is a catch-all for such non-terminating behaviour.Issues