jnp.where is not working with Metal GPU backend XlaRuntimeError Traceback (most recent call last) Cell In[49], line 4 1 arr = jnp.array( [7, 8, 9]) 3 # Find indices where the condition is True ----> 4 indices = jnp.where(arr > 1) 6 print(indices) XlaRuntimeError: UNKNOWN Machine Learning & AI General tensorflow-metal 7 2 1.1k Jun ’23