Run jax doctests in jax CI

### Description

There are some complex code examples added in #1614, but they don't run as part of the doctests because we explicitly ignore `pytensor.link` module (I don't remember if because of the optional dependencies or because something else failed).

We should run the doctest explicitly on pytensor.link.jax on the  JAX CI