Comment by mikepapadim

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