-
Notifications
You must be signed in to change notification settings - Fork 39
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
LightningQubit
VJP support
#644
Conversation
[sc-59045] |
Codecov ReportAll modified and coverable lines are covered by tests ✅
Additional details and impacted files@@ Coverage Diff @@
## master #644 +/- ##
==========================================
+ Coverage 98.39% 98.66% +0.27%
==========================================
Files 172 172
Lines 25138 24376 -762
==========================================
- Hits 24734 24051 -683
+ Misses 404 325 -79 ☔ View full report in Codecov by Sentry. |
Lots of test failures in Pennylane due to lack of VJP support in Lightning. I am skipping such tests for now, but they should stop being skipped once this PR is merged. See the files changed in PL #5434 to find which tests need to be updated. |
TODO in pennylane: Uncomment interface tests using |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Nice job! Thanks!
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks good to me, nice work @mudit2812 !
Context:
This PR adds adjoint VJP computation support to
lightning.qubit
with the new API.Description of the Change:
LightningQubit.compute_vjp
,supports_vjp
, andexecute_and_compute_vjp
methods along with appropriate helper functions.tests/test_vjp.py
to stop skipping new API devices.tests/new_api/test_device.py
.Benefits:
LightningQubit
can now set thedevice_vjp
argument to automatically use the VJP pipeline for differentiation.Possible Drawbacks:
Related GitHub Issues: