flax

1dc3d541 - [flax] Add messages to some assertions in `pool` to help with debugging.