Add Jax WaldRV (invgauss)

[FredericWantiez]

This adds a Jax implementation of WaldRV to close #1329. I've added a cdf test but for scipy invgauss means lam=1 (here), I should probably add a few more

[codecov]

Codecov Report

Merging #1435 (059fe55) into main (2f9a9e6) will increase coverage by 0.02%.
The diff coverage is 100.00%.

❗ Current head 059fe55 differs from pull request most recent head a3dc641. Consider uploading reports for the commit a3dc641 to get more accurate results

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #1435      +/-   ##
==========================================
+ Coverage   74.72%   74.75%   +0.02%     
==========================================
  Files         194      194              
  Lines       49877    49912      +35     
  Branches    10549    10553       +4     
==========================================
+ Hits        37273    37314      +41     
+ Misses      10274    10270       -4     
+ Partials     2330     2328       -2     

Impacted Files	Coverage Δ
aesara/link/jax/dispatch/random.py	100.00% <100.00%> (ø)
aesara/scalar/__init__.py	100.00% <0.00%> (ø)
aesara/scalar/basic.py	79.16% <0.00%> (+0.12%)	⬆️
aesara/tensor/math.py	90.68% <0.00%> (+0.26%)	⬆️
aesara/tensor/__init__.py	96.29% <0.00%> (+0.74%)	⬆️

[brandonwillard]

Just recommended some updates to the docstring; otherwise, looks good! Many thanks, @FredericWantiez!