Downstreaming Aesara PR 1284: "Fix the JAX RandomVariable dispatcher"

[twiecki]

Downstreaming aesara-devs/aesara#1284. PR port done by downstream_pr.sh script.

[codecov-commenter]

Codecov Report

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

Project coverage is 74.40%. Comparing base (039ed1c) to head (571b106).
Report is 825 commits behind head on main.

Files with missing lines	Patch %	Lines
pytensor/link/jax/dispatch/shape.py	93.33%	1 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main     #103      +/-   ##
==========================================
+ Coverage   74.32%   74.40%   +0.07%     
==========================================
  Files         177      179       +2     
  Lines       49078    49227     +149     
  Branches    10416    10420       +4     
==========================================
+ Hits        36479    36628     +149     
- Misses      10293    10294       +1     
+ Partials     2306     2305       -1     

Files with missing lines	Coverage Δ
pytensor/compile/mode.py	84.47% <ø> (ø)
pytensor/link/jax/dispatch/random.py	100.00% <100.00%> (+2.63%)	⬆️
pytensor/tensor/random/rewriting/__init__.py	100.00% <100.00%> (ø)
pytensor/tensor/random/rewriting/basic.py	93.54% <ø> (ø)
pytensor/tensor/random/rewriting/jax.py	100.00% <100.00%> (ø)
pytensor/link/jax/dispatch/shape.py	88.46% <93.33%> (+1.97%)	⬆️