jax

18698a1f - [shape_poly] Add support for jnp.split