change r2.5 to use jax stable version 0.4.33

pytorch / xla

Enabling PyTorch on XLA Devices (e.g. Google TPU)

https://pytorch.org/xla

Other

2.49k stars 480 forks source link

Closed ManfeiBai closed 1 month ago

ManfeiBai commented 1 month ago

follow-up PR to update OpenXLA-commit to the same one used by JAX stable version: https://github.com/pytorch/xla/pull/8229

suggest to combine these two PRs together in the next time to do similar thing