dire_jax
tests
