Update pennylane version (#2249)

**Context:**
After two bumps of jax version in both Catalyst and Pennylane, here we
update pennylane's pinned version back to the latest dev

**Description of the Change:**

**Benefits:**

**Possible Drawbacks:**

**Related GitHub Issues:**

---------

Co-authored-by: Hong-Sheng Zheng <[email protected]>
Co-authored-by: Ali Asadi <[email protected]>
Co-authored-by: paul0403 <[email protected]>

Author: 4 people

.dep-versions

@@ -10,9 +10,7 @@ enzyme=v0.0.203
 
 # For a custom PL version, update the package version here and at
 # 'doc/requirements.txt'
-# TODO: uncomment and update to latest version of pennylane
-#       after https://github.com/PennyLaneAI/pennylane/pull/8525 is merged.
-pennylane @ git+https://github.com/PennyLaneAI/pennylane@bump-jax-api-hashability
+pennylane=0.44.0-dev42
 
 # For a custom LQ/LK version, update the package version here and at
 # 'doc/requirements.txt'

.github/workflows/build-wheel-linux-arm64.yaml

@@ -478,11 +478,6 @@ jobs:
       run: |
         python${{ matrix.python_version }} -m pip install oqc-qcaas-client
 
-    - name: Install PennyLane
-      run: |
-        # Install the specific pennylane version required
-        python${{ matrix.python_version }} -m pip install git+https://github.com/PennyLaneAI/pennylane@bump-jax-api-hashability
-
     - name: Install Catalyst
       run: |
         python${{ matrix.python_version }} -m pip install dist/*.whl --extra-index-url https://test.pypi.org/simple

.github/workflows/build-wheel-linux-x86_64.yaml

@@ -492,11 +492,6 @@ jobs:
         python${{ matrix.python_version }} -m pip install PennyLane-Lightning-Kokkos
         python${{ matrix.python_version }} -m pip install 'amazon-braket-pennylane-plugin>=1.31.0'
 
-    - name: Install PennyLane
-      run: |
-        # Install the specific pennylane version required
-        python${{ matrix.python_version }} -m pip install git+https://github.com/PennyLaneAI/pennylane@bump-jax-api-hashability
-
     - name: Install OQC client
       if: matrix.python_version != '3.13'
       run: |

.github/workflows/build-wheel-macos-arm64.yaml

@@ -473,11 +473,6 @@ jobs:
         python${{ matrix.python_version }} -m pip install PennyLane-Lightning-Kokkos
         python${{ matrix.python_version }} -m pip install 'amazon-braket-pennylane-plugin>1.27.1'
 
-    - name: Install PennyLane
-      run: |
-        # Install the specific pennylane version required
-        python${{ matrix.python_version }} -m pip install git+https://github.com/PennyLaneAI/pennylane@bump-jax-api-hashability
-
     - name: Install OQC client
       if: matrix.python_version != '3.13'
       run: |

Makefile

@@ -117,8 +117,6 @@ frontend:
 	# versions of a package with the same version tag (e.g. 0.38-dev0).
 	$(PYTHON) -m pip uninstall -y pennylane
 	$(PYTHON) -m pip install -e . --extra-index-url https://test.pypi.org/simple $(PIP_VERBOSE_FLAG)
-	# TODO: remove after https://github.com/PennyLaneAI/pennylane/pull/8525 is merged.
-	$(PYTHON) -m pip install git+https://github.com/PennyLaneAI/pennylane@bump-jax-api-hashability
 	rm -r frontend/pennylane_catalyst.egg-info
 
 .PHONY: mlir llvm stablehlo enzyme dialects runtime oqc

doc/releases/changelog-dev.md

@@ -72,13 +72,6 @@
 * Resource tracking now tracks calls to `SetState` and `SetBasisState`, and can report results
   that include `qml.StatePrep` operations.
   [(#2230)](https://github.com/PennyLaneAI/catalyst/pull/2230)
-  
-* Remove the hardcoded list of runtime operations in the frontend.
-  This will allow arbitrary PL gates to be represented without hyperparameters in MLIR.
-  For gates that do not have a QIR representation, a runtime error will be raised at execution.
-  Users can still decompose these gates via `qml.transforms.decompose`
-  when both capture and graph-decomposition are enabled.
-  [(#2215)](https://github.com/PennyLaneAI/catalyst/pull/2215)
 
 * `qml.PCPhase` can be compiled and executed with capture enabled.
   [(#2226)](https://github.com/PennyLaneAI/catalyst/pull/2226)
@@ -147,6 +140,9 @@
 * The plxpr transform `pl_map_wires` has been removed along with its test.
   [(#2220)](https://github.com/PennyLaneAI/catalyst/pull/2220)
 
+* The JAX version used by Catalyst has been updated to 0.7.0.
+  [(#2131)](https://github.com/PennyLaneAI/catalyst/pull/2131)
+
 * (Compiler integrators only) The versions of LLVM/Enzyme/stablehlo used by Catalyst have been
   updated. Enzyme now targets `v0.0.203` with the build target `EnzymeStatic-22`, and the nanobind
   requirement for the latest LLVM has been updated to version 2.9.

doc/requirements.txt

@@ -33,4 +33,4 @@ lxml_html_clean
 --extra-index-url https://test.pypi.org/simple/
 pennylane-lightning-kokkos==0.44.0-dev11
 pennylane-lightning==0.44.0-dev11
-git+https://github.com/PennyLaneAI/pennylane@bump-jax-api-hashability
+pennylane==0.44.0-dev42