Many components for a jax for java are already tested in https://github.com/beehive-lab/jitllm