jax-rt is not only in runtime needed