jax

42d97c87 - Merge pull request #35576 from jakevdp:pyrefly-opt-einsum