catalyst

catalyst copied to clipboard

This issue reflexes the [fixme comment](https://github.com/PennyLaneAI/catalyst/blob/d6d4d8e0526609d36f2ae178bdaf03342850e9bc/frontend/catalyst/api_extensions/control_flow.py#L538) in the quantum tracing pipeline for the conditional primitive. The core of the problem is: we re-trace Jaxpr program in order to unify the...

sergei-mironov

#### Issue description ~~It looks like ``qjit`` is not well compitable with ``functools.partial``.~~ ~~The following example triggers ``TypeError: fn() missing 1 required positional argument: 'y'`` error.~~ The main reason is...

tzunghanjuang

`jax.grad` (and `catalyst.grad`) sometimes randomly fails on a qjit-ted function containing a qnode: ```python dev = qml.device("lightning.qubit", wires=1) @qjit def cost(x): @qml.qnode(dev) def circuit(x): qml.PauliX(wires=0) #qml.RZ(0, wires=0) return qml.probs() return...

paul0403

Consider eliminating redundant "shots" parameters in device back end API

1

#### Issue description As a general suggestion on your (C++ device back end) API, in `Catalyst::Runtime::QuantumDevice`, as someone implementing a simulator device back end over at [pennylane-qrack](https://github.com/unitaryfund/pennylane-qrack), I notice that...

WrathfulSpatula

Add support to AutoGraph for

1

**Context:** https://github.com/PennyLaneAI/catalyst/pull/717 added support for converting in-place array updates (`arr[i] = x`) into the equivalent JAX traceable code (`arr.at[i].set(x)`). This change extends that support to operator assignment array updates. **Description...

Spencer-Comin

[`--buffer-results-to-out-params`](https://mlir.llvm.org/docs/Passes/#-buffer-results-to-out-params) is a pass in upstream MLIR that will modify functions that return memrefs to functions that receive memrefs parameters to denote the return values. In Catalyst we use this...

erick-xanadu

**Context** Catalyst uses a source-to-source transformation package called [AutoGraph](https://docs.pennylane.ai/projects/catalyst/en/stable/dev/autograph.html), which allows users to write regular Python code that is automagically transformed into JAX-style traceable code. For example, the following for...

dime10

Find a better way to link blas routines from jaxlib

2

Currently catalyst links to the blas library shipped with scipy, not jaxlib, because jaxlib does not ship their blas library with a .so shared object file. This results in when...

paul0403

Migrating to One-Shot Bufferization

12

**Context:** MLIR removed old bufferization (`Conversion-Based bufferization`) passes and introduced new `One-Shot-Bufferization` pass. [[Link]](https://mlir.llvm.org/docs/Bufferization/) Note that `Ownership-based Buffer Deallocation` will be handled in a different PR. - ~Only update llvm...

tzunghanjuang

Create a finite state machine analysis pass to propagate Pauli basis states, and use it to disentangle CNOTs

2

**Context:** In circuits that mostly involve the six Pauli eigenstates, the effect of gates can be described by a simple six-state finite state machine. This allows us to deduce their...

paul0403