jax

0dbda849 - lax.dynamic_slice: avoid negative index correction for unsigned indices