jax

d2f937e2 - Make jax.Arrays a necessary part of the cycle in the GC guard test

Make jax.Arrays a necessary part of the cycle in the GC guard test Otherwise, the cycle can be broken by clearing the references of the helper objects, at which points the deallocation of arrays proceeds through regular reference counting (and does not trigger logs!). I have not verified that this is what happens, but the test has been mysteriously failing under a number of configurations and this seems to fix it. I added a note to the garbage collection guard to clarify that it's not guaranteed to report all cycles. PiperOrigin-RevId: 708320953