jax.numpy indexing: lower to dynamic_slice for more cases#15377

Merged

copybara-service[bot] merged 1 commit intojax-ml:mainfrom jakevdp:gather-slice

Apr 17, 2023

+139-34

Commits on Apr 17, 2023

jax.numpy indexing: lower to dynamic_slice for more cases
jakevdp
committed