Capture a qnode into jaxpr

[albi3ro]

Context:

Description of the Change:

Benefits:

Possible Drawbacks:

Related GitHub Issues:

[sc-61301]

[codecov]

Codecov Report

Attention: Patch coverage is 98.36066% with 1 line in your changes missing coverage. Please review.

Project coverage is 99.66%. Comparing base (92ce59d) to head (aab3442).
Report is 259 commits behind head on master.

Files with missing lines	Patch %	Lines
pennylane/capture/capture_qnode.py	98.18%	1 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##           master    #5708      +/-   ##
==========================================
- Coverage   99.67%   99.66%   -0.01%     
==========================================
  Files         413      414       +1     
  Lines       39250    39017     -233     
==========================================
- Hits        39123    38887     -236     
- Misses        127      130       +3     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[dwierichs]

Nice work in those tests @albi3ro
Excited that this integration requires so little effort 🎉

We could think about modifying the printing of qnode jaxprs in a future PR, to enhance readability :)

The names of existing files suggests to rename the new file to tests/capture/test_qnode.py 🙃

[dime10]

Love it, this looks great 💯

[dwierichs]

Thanks @albi3ro, looking forward to this going in!

I'm approving, conditioned on a conscious decision about obtaining the shots in the beginning of qnode_call: Should we always call _get_device_shots, and cast to a Shots object even if it is one already? It's not crucial, but deserves a thought, I think :)