mirror of
https://github.com/ROCm/jax.git
synced 2025-04-16 11:56:07 +00:00

For dynamic shapes experimentation and normal debugging, `python_pjit` still exists so that problem doesn't exist which makes us free to remove these 2 flags. I am leaving pmap's flag alone for now. PiperOrigin-RevId: 522602754