jax

f5cffd72 - delete more dead index_take code