Convolution Example - Jax Typing CST Test