mirror of
https://github.com/ROCm/jax.git
synced 2025-04-15 19:36:06 +00:00

This change is needed to add testing of int2/uint2 dtypes via bazel in presubmit (see https://github.com/jax-ml/jax/pull/21395). PiperOrigin-RevId: 735895293