multiple fixes

Author: erick-xanadu

frontend/catalyst/pipelines.py

@@ -174,6 +174,7 @@ def get_enforce_runtime_invariants_stage(_options: CompileOptions) -> List[str]:
         # But qnodes targeting other backends may choose to lower
         # this into something else.
         "builtin.module(inline)",
+        "split-multiple-tapes",
         "inline-nested-module",
     ]
     return enforce_runtime_invariants

frontend/catalyst/qfunc.py

@@ -132,8 +132,9 @@ def __call__(self, *args, **kwargs):
         debug_info = kwargs.pop("debug_info", None)
 
         import jax
+        from functools import partial
 
-        @jax.jit
+        @partial(jax.jit, static_argnums=static_argnums)
         def _eval_quantum(*args, **kwargs):
             closed_jaxpr, out_type, out_tree, out_tree_exp = trace_quantum_function(
                 self.func,