skip jax test if jax not installed

Author: matthiasdiener

test/test_jax.py

@@ -23,6 +23,8 @@
 import pytest
 import numpy as np
 import pytato as pt
+
+pytest.importorskip("jax")
 from jax.config import config
 config.update("jax_enable_x64", True)