diff --git a/.github/workflows/braket-latest-latest.yml b/.github/workflows/braket-latest-latest.yml
index 41860db..d115915 100644
--- a/.github/workflows/braket-latest-latest.yml
+++ b/.github/workflows/braket-latest-latest.yml
@@ -16,9 +16,8 @@ env:
   PLUGIN_BRANCH: main
   PLUGIN_PACKAGE: amazon-braket-pennylane-plugin
   PENNYLANE_BRANCH: main
-  TF_VERSION: 2.12.0
-  TORCH_VERSION: 2.0.0+cpu
-  JAX_VERSION: 0.4.28
+  TORCH_VERSION: 2.9.1+cpu
+  JAX_VERSION: 0.7.1
 
 
 jobs:
@@ -39,10 +38,6 @@ jobs:
         with:
           python-version: "3.12"
             
-      - name: Install TF
-        run: |
-          pip install tensorflow~=$TF_VERSION keras~=$TF_VERSION
-
       - name: Install JAX
         run: |
           pip install jax==$JAX_VERSION jaxlib==$JAX_VERSION
diff --git a/.github/workflows/braket-latest-rc.yml b/.github/workflows/braket-latest-rc.yml
index eaa76f5..d38fcd6 100644
--- a/.github/workflows/braket-latest-rc.yml
+++ b/.github/workflows/braket-latest-rc.yml
@@ -16,9 +16,8 @@ env:
   PLUGIN_BRANCH: main
   PLUGIN_PACKAGE: amazon-braket-pennylane-plugin
   PENNYLANE_BRANCH: main
-  TF_VERSION: 2.12.0
-  TORCH_VERSION: 2.0.0+cpu
-  JAX_VERSION: 0.4.28
+  TORCH_VERSION: 2.9.1+cpu
+  JAX_VERSION: 0.7.1
 
 
 jobs:
@@ -39,10 +38,6 @@ jobs:
         with:
           python-version: "3.12"
             
-      - name: Install TF
-        run: |
-          pip install tensorflow~=$TF_VERSION keras~=$TF_VERSION
-
       - name: Install JAX
         run: |
           pip install jax==$JAX_VERSION jaxlib==$JAX_VERSION
diff --git a/.github/workflows/braket-latest-stable.yml b/.github/workflows/braket-latest-stable.yml
index d92a5df..7d84367 100644
--- a/.github/workflows/braket-latest-stable.yml
+++ b/.github/workflows/braket-latest-stable.yml
@@ -16,9 +16,8 @@ env:
   PLUGIN_BRANCH: main
   PLUGIN_PACKAGE: amazon-braket-pennylane-plugin
   PENNYLANE_BRANCH: main
-  TF_VERSION: 2.12.0
-  TORCH_VERSION: 2.0.0+cpu
-  JAX_VERSION: 0.4.28
+  TORCH_VERSION: 2.9.1+cpu
+  JAX_VERSION: 0.7.1
 
 
 jobs:
@@ -37,12 +36,8 @@ jobs:
       - name: Set up Python
         uses: actions/setup-python@v2
         with:
-          python-version: "3.11"
+          python-version: "3.12"
             
-      - name: Install TF
-        run: |
-          pip install tensorflow~=$TF_VERSION keras~=$TF_VERSION
-
       - name: Install JAX
         run: |
           pip install jax==$JAX_VERSION jaxlib==$JAX_VERSION
diff --git a/.github/workflows/braket-stable-latest.yml b/.github/workflows/braket-stable-latest.yml
index d456a6c..3d1ff17 100644
--- a/.github/workflows/braket-stable-latest.yml
+++ b/.github/workflows/braket-stable-latest.yml
@@ -16,9 +16,8 @@ env:
   PLUGIN_BRANCH: main
   PLUGIN_PACKAGE: amazon-braket-pennylane-plugin
   PENNYLANE_BRANCH: main
-  TF_VERSION: 2.12.0
-  TORCH_VERSION: 2.0.0+cpu
-  JAX_VERSION: 0.4.28
+  TORCH_VERSION: 2.9.1+cpu
+  JAX_VERSION: 0.7.1
 
 
 jobs:
@@ -39,10 +38,6 @@ jobs:
         with:
           python-version: "3.12"
             
-      - name: Install TF
-        run: |
-          pip install tensorflow~=$TF_VERSION keras~=$TF_VERSION
-
       - name: Install JAX
         run: |
           pip install jax==$JAX_VERSION jaxlib==$JAX_VERSION
diff --git a/.github/workflows/braket-stable-stable.yml b/.github/workflows/braket-stable-stable.yml
index 5d3f910..f2487df 100644
--- a/.github/workflows/braket-stable-stable.yml
+++ b/.github/workflows/braket-stable-stable.yml
@@ -16,9 +16,8 @@ env:
   PLUGIN_BRANCH: main
   PLUGIN_PACKAGE: amazon-braket-pennylane-plugin
   PENNYLANE_BRANCH: main
-  TF_VERSION: 2.12.0
-  TORCH_VERSION: 2.0.0+cpu
-  JAX_VERSION: 0.4.28
+  TORCH_VERSION: 2.9.1+cpu
+  JAX_VERSION: 0.7.1
 
 
 jobs:
@@ -37,12 +36,8 @@ jobs:
       - name: Set up Python
         uses: actions/setup-python@v2
         with:
-          python-version: "3.11"
+          python-version: "3.12"
             
-      - name: Install TF
-        run: |
-          pip install tensorflow~=$TF_VERSION keras~=$TF_VERSION
-
       - name: Install JAX
         run: |
           pip install jax==$JAX_VERSION jaxlib==$JAX_VERSION
diff --git a/.github/workflows/ionq-latest-stable.yml b/.github/workflows/ionq-latest-stable.yml
index 740fa24..fc27d0f 100644
--- a/.github/workflows/ionq-latest-stable.yml
+++ b/.github/workflows/ionq-latest-stable.yml
@@ -35,7 +35,7 @@ jobs:
       - name: Set up Python
         uses: actions/setup-python@v2
         with:
-          python-version: "3.11"
+          python-version: "3.12"
 
       - name: Install requirements
         run: |
diff --git a/.github/workflows/ionq-stable-stable.yml b/.github/workflows/ionq-stable-stable.yml
index 8ec2bc3..73ccaa7 100644
--- a/.github/workflows/ionq-stable-stable.yml
+++ b/.github/workflows/ionq-stable-stable.yml
@@ -35,7 +35,7 @@ jobs:
       - name: Set up Python
         uses: actions/setup-python@v2
         with:
-          python-version: "3.11"
+          python-version: "3.12"
 
       - name: Install requirements
         run: |
diff --git a/.github/workflows/qiskit-latest-stable.yml b/.github/workflows/qiskit-latest-stable.yml
index 9e6cf84..5bb2829 100644
--- a/.github/workflows/qiskit-latest-stable.yml
+++ b/.github/workflows/qiskit-latest-stable.yml
@@ -35,7 +35,7 @@ jobs:
       - name: Set up Python
         uses: actions/setup-python@v2
         with:
-          python-version: "3.11"
+          python-version: "3.12"
 
       - name: Install requirements
         run: |
diff --git a/.github/workflows/qiskit-stable-stable.yml b/.github/workflows/qiskit-stable-stable.yml
index a01a8e3..5ca4a51 100644
--- a/.github/workflows/qiskit-stable-stable.yml
+++ b/.github/workflows/qiskit-stable-stable.yml
@@ -35,7 +35,7 @@ jobs:
       - name: Set up Python
         uses: actions/setup-python@v2
         with:
-          python-version: "3.11"
+          python-version: "3.12"
 
       - name: Install requirements
         run: |
diff --git a/compile.py b/compile.py
index 89af071..7071a33 100644
--- a/compile.py
+++ b/compile.py
@@ -39,17 +39,13 @@
             "--device=braket.local.qubit --tb=short --skip-ops -k 'not Sample and not no_0_shots'",
         ],
         "tests_loc": "test/unit_tests",
-        "additional_setup": dedent("""
-            - name: Install TF
-              run: |
-                pip install tensorflow~=$TF_VERSION keras~=$TF_VERSION
-                                   
+        "additional_setup": dedent("""                                   
             - name: Install JAX
               run: |
                 pip install jax==$JAX_VERSION jaxlib==$JAX_VERSION"""
                                    
         ),
-        "additional_env_vars": "TF_VERSION: 2.12.0\n  TORCH_VERSION: 2.0.0+cpu\n  JAX_VERSION: 0.4.28",
+        "additional_env_vars": "TORCH_VERSION: 2.9.1+cpu\n  JAX_VERSION: 0.7.1",
         "no_deprecation_error": True,
     },
 ]
diff --git a/workflow-template-stable.yml b/workflow-template-stable.yml
index 37b8f36..8f752f7 100644
--- a/workflow-template-stable.yml
+++ b/workflow-template-stable.yml
@@ -40,7 +40,7 @@ jobs:
       - name: Set up Python
         uses: actions/setup-python@v2
         with:
-          python-version: "3.11"
+          python-version: "3.12"
 
 {%- if additional_setup is defined %}
       {{ additional_setup | indent(6, True) }}