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%)	⬆️