JIT Compilation Example - Jax Typing CST Test