Warning from metrics.IoU with Jax backend

[t-kalinowski]

This line causes Jax to emit a warning:

keras/keras/src/metrics/metrics_utils.py

Line 679 in 6d26efb

indices = ops.cast(indices, dtype="int64")

/home/tomasz/.cache/R/reticulate/uv/cache/archive-v0/KYN5vaJzRb8cASt2E8jyn/lib/python3.11/site-packages/keras/src/backend/jax/core.py:80: 
UserWarning: Explicitly requested dtype int64 requested in asarray is not available, 
and will be truncated to dtype int32. To enable more dtypes, set the jax_enable_x64 
configuration option or the JAX_ENABLE_X64 shell environment variable. 
See https://github.com/jax-ml/jax#current-gotchas for more.
  return jnp.asarray(x, dtype=dtype)

[sonali-kumari1]

Hi @t-kalinowski -

Could you please share a sample code to reproduce this issue? Thanks!