Noisy patterns disappearing between 0.4.28 and 0.5.3

[r-toroxel]

thanks for the great tool. this is some art i make with it. but one question.

why do these nice noisy infinity patterns (those noisy dots on top of the smooth pattern, all of non finite value) (picture made with jax/jaxlib version 0.4.28) stop somewhere up to jax/jaxlib version 0.5.3? please point me to any commits or ideas what change may have broken that.
may have been a numerical instability bug before.

all i can say is that i could turn it off by using float64 instead of default float32. but 0.5.3 still uses float32 yet those patterns disappear.

[mattjj]

Awesome art!

There's not really enough for us to go on here to diagnose what changed. Were there random values involved? Can you share a simple repro? You might also consider bisecting against the jax/jaxlib version to pin things down a bit more, though with a simple repro we could bisect to the exact change using google infra.

[r-toroxel]

Thx!

Will produce a repro when i get to it.