mirror of
https://github.com/ROCm/jax.git
synced 2025-04-24 00:56:05 +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