jax

d46202e4 - Add some tracemes in py_array to make slow device put debugging easier.