Bump jax to 0.7.1#2134

rniczh wants to merge 6 commits intorniczh/bump-jax-to-0.7.0PennyLaneAI/catalyst:rniczh/bump-jax-to-0.7.0from

rniczh/bump-jax-to-0.7.1PennyLaneAI/catalyst:rniczh/bump-jax-to-0.7.1Copy head branch name to clipboard