Jax tests: skip if jax is not available

15 jobs for sampling_dtypes_in_sumops in 21 minutes and 20 seconds (queued for 6 seconds)