live_arrays() returns confusing duplicate items.

[pipme]

Hi, here's an example with Jax 0.4.19:

import jax 
import jax.numpy as jnp

a = jnp.ones((3,3))

def print_buffers():   
    client = jax.lib.xla_bridge.get_backend()
    lbs = [(lb.shape, lb.unsafe_buffer_pointer()) for lb in client.live_arrays()]
    for idx, lb in enumerate(lbs):
        print(
            f"lbs[{idx}]: {lb}"
        )


print_buffers()
print("-"*20)
#import gc
#import time
#gc.collect()
#time.sleep(1)
print_buffers()

The output of the above is:

lbs[0]: ((3, 3), 4736204624)
--------------------
lbs[0]: ((3, 3), 4736204624)
lbs[1]: ((3, 3), 4736204624)

There is only one live array but why at the second call it prints two live arrays with identical unsafe_buffer_pointer. (I suppose unsafe_buffer_pointer is the actual memory address of the array)

Thanks!