test(use_jax): add fallback assertion + skip happy path when jax missing

test(use_jax): add fallback assertion + skip happy path when jax missing #3322