jax

17b5fe11 - add test for issue #553