jax

263eed93 - [mosaic_gpu] The `barrier` operand to `async_load` is now optional on pre-Hopper GPUs