JAX: Implement geometric RV sampling

[Smit-create]

Fixes #1323 Here are a few important guidelines and requirements to check before your PR can be merged:

• [x] There is an informative high-level description of the changes.
• [x] The description and/or commit message(s) references the relevant GitHub issue(s).
• [x] pre-commit is installed and set up.
• [x] The commit messages follow these guidelines.
• [x] The commits correspond to relevant logical changes, and there are no commits that fix changes introduced by other commits in the same branch/BR.
• [x] There are tests covering the changes introduced in the PR.

[codecov[bot]]

Codecov Report

Merging #1444 (552346f) into main (4c74650) will increase coverage by 0.00%. The diff coverage is 100.00%.

Additional details and impacted files

[](https://codecov.io/gh/aesara-devs/aesara/pull/1444?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=aesara-devs) ```diff @@ Coverage Diff @@ ## main #1444 +/- ## ======================================= Coverage 74.81% 74.81% ======================================= Files 194 194 Lines 49995 50007 +12 Branches 10572 10572 ======================================= + Hits 37402 37414 +12 Misses 10267 10267 Partials 2326 2326 ``` | [Impacted Files](https://codecov.io/gh/aesara-devs/aesara/pull/1444?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=aesara-devs) | Coverage Δ | | |---|---|---| | [aesara/link/jax/dispatch/random.py](https://codecov.io/gh/aesara-devs/aesara/pull/1444?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=aesara-devs#diff-YWVzYXJhL2xpbmsvamF4L2Rpc3BhdGNoL3JhbmRvbS5weQ==) | `100.00% <100.00%> (ø)` | |

[rlouf]

Thank you @Smit-create!