jax

588c2812 - [Mosaic TPU] No longer implicitly trigger single-buffering when running out of VMEM

[Mosaic TPU] No longer implicitly trigger single-buffering when running out of VMEM It's just more predictable to OOM and let the user explicitly opt into single buffering using `pipeline_mode=pl.Buffered(1)`. We bump the version number so that the old artifacts still get the relaxed treatment. PiperOrigin-RevId: 827487842