test: skip jax-backed tests when jax is absent (Python matrix)

test: skip jax-backed tests when jax is absent (Python matrix) #2153