skip test_vit_b16 until https://github.com/jax-ml/jax/issues/25066 is…

skip test_vit_b16 until https://github.com/jax-ml/jax/issues/25066 is… #21