[Capture] Qnode natively executes jaxpr on the device

PennyLaneAI / pennylane

PennyLane is a cross-platform Python library for quantum computing, quantum machine learning, and quantum chemistry. Train a quantum computer the same way as a neural network.

https://pennylane.ai

Apache License 2.0

2.38k stars 607 forks source link

[Capture] Qnode natively executes jaxpr on the device #6596

Open albi3ro opened 1 week ago

albi3ro commented 1 week ago

Context:

Though we can now capture entire workflows and execute jaxpr on default qubit, we are still falling back to the old execution pipeline for execution.

Description of the Change:

Adds a new implementation to the qnode primitive that sends the qfunc jaxpr to the device.

It also adds a couple minor fixes to DefaultQubitInterpreter.

Benefits:

Possible Drawbacks:

Related GitHub Issues:

[sc-78422]

github-actions[bot] commented 1 week ago

Hello. You may have forgotten to update the changelog! Please edit doc/releases/changelog-dev.md with:

A one-to-two sentence description of the change. You may include a small working example for new features.
A link back to this PR.
Your name (or GitHub username) in the contributors section.

codecov[bot] commented 1 day ago

Codecov Report

All modified and coverable lines are covered by tests :white_check_mark:

Project coverage is 99.63%. Comparing base (aae481a) to head (9b9d939). Report is 4 commits behind head on master.

Additional details and impacted files

```diff @@ Coverage Diff @@ ## master #6596 +/- ## ========================================== - Coverage 99.65% 99.63% -0.02% ========================================== Files 455 455 Lines 42881 42937 +56 ========================================== + Hits 42732 42782 +50 - Misses 149 155 +6 ```

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.