jax

1014bf3d - Make IndivisibleError more understandable by reporting the correct dim_size.