jax

6c0ac7a5 - Do a pvary in dynamic_slice_transpose_rule so that the `zeros` are varying with the correct vma as the operands were.