Simplify dispatch of JAX random variables by handling rng split automatically

[educhesne]

Description

Move the jax rng splits from the jax_sample_fn of each RandomVariable dispatch to jax_funcify_RandomVariable

Related Issue

• Closes Simplify dispatch of JAX random variables by handling rng split automatically #1204
• Related to #

Checklist

• Checked that the pre-commit linting/style checks pass
• Included tests that prove the fix is effective or that the new feature works
• Added necessary documentation (docstrings and/or example notebooks)
• If you are a pro: each commit corresponds to a relevant logical change

Type of change

• New feature / enhancement
• Bug fix
• Documentation
• Maintenance
• Other (please specify): refacto

---

📚 Documentation preview 📚: https://pytensor--1315.org.readthedocs.build/en/1315/

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 81.98%. Comparing base (95ce102) to head (98443f1).
Report is 149 commits behind head on main.

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #1315      +/-   ##
==========================================
- Coverage   81.99%   81.98%   -0.02%     
==========================================
  Files         188      188              
  Lines       48508    48474      -34     
  Branches     8672     8672              
==========================================
- Hits        39773    39739      -34     
  Misses       6583     6583              
  Partials     2152     2152              

Files with missing lines	Coverage Δ
pytensor/link/jax/dispatch/random.py	92.72% <100.00%> (-0.98%)	⬇️

🚀 New features to boost your workflow:

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

[ricardoV94]

Thanks @educhesne, much cleaner now