I am experiencing a weird issue when indexing a Jax array using a list.
If I place a debugger in the middle of my code, I have the following:

This array are created by convering a numpy array.

However, when I try this in a new instance of Python, I have the correct behavior:
[

What is it happening?