jax

479f256a - py3 iterator compatibility