jax

48934dc9 - Implement cross product and test cases