[Capture] handle grad and jacobian

[albi3ro]

Context:

Handles the grad primitive from pennylane and adds a lowering for it.

Description of the Change:

Benefits:

Possible Drawbacks:

Related GitHub Issues:

[sc-100560]

[github-actions]

Hello. You may have forgotten to update the changelog!
Please edit doc/releases/changelog-dev.md on your branch 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]

Codecov Report

❌ Patch coverage is 71.83099% with 20 lines in your changes missing coverage. Please review.
✅ Project coverage is 97.28%. Comparing base (a007d47) to head (81e46d2).

Files with missing lines	Patch %	Lines
frontend/catalyst/jax_primitives.py	25.00%	12 Missing ⚠️
frontend/catalyst/jax_primitives_utils.py	89.79%	3 Missing and 2 partials ⚠️
frontend/catalyst/from_plxpr/from_plxpr.py	50.00%	3 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #2078      +/-   ##
==========================================
- Coverage   97.48%   97.28%   -0.21%     
==========================================
  Files          91       91              
  Lines       10594    10622      +28     
  Branches      990      993       +3     
==========================================
+ Hits        10328    10334       +6     
- Misses        211      230      +19     
- Partials       55       58       +3     

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

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.