#1 interface JAX.IIterable, some fixes