mirror of
https://github.com/ROCm/jax.git
synced 2025-04-18 21:06:06 +00:00

(I had missed that the upstream pass exists!) Fixes https://github.com/google/jax/issues/16649 PiperOrigin-RevId: 548192839
(I had missed that the upstream pass exists!) Fixes https://github.com/google/jax/issues/16649 PiperOrigin-RevId: 548192839