jax_template added