Vectorized Operations Example - Jax Typing CST Test