jax

21a3d50d - Skip test_get_vmap for known _get_vmap slice-only bug

Skip test_get_vmap for known _get_vmap slice-only bug Use self.skipTest() to skip when hitting buggy case: - No integer indexers (only slices) - ref_bdim != slice_bdim Bug: jax-ml/jax#33309 Fix: cherry-pick upstream ce6f33d55