jax

4aee3810 - Disallow unreduced inputs for all primitives except those that implement the unreduced rule. Currently that's only `add`.