jax

c9988ab4 - Limit matrix to a single combination for faster tests