Added Jax generated code and test project