feat: gather iota indexing optimizations#1903

wsmoses merged 1 commit intomainEnzymeAD/Enzyme-JAX:mainfrom

ap/scatter_gather_with_iota_indexingEnzymeAD/Enzyme-JAX:ap/scatter_gather_with_iota_indexingCopy head branch name to clipboard