jax

16712b51 - Ensure that the two offsets of a dynamic_slice have the same dtype regardless