jax

7239d15f - [Mosaic GPU] Remove `TraversalOrder` from `traverse_op`.

[Mosaic GPU] Remove `TraversalOrder` from `traverse_op`. We only use it in one place now, and therefore no longer need this parametrization. PiperOrigin-RevId: 826556517