jax

e150368a - Relax `TpuInfo.is_matmul_supported` for TPU7x (Ironwood) to match flexibility already built in HW.