jax

1c2e635f - more general/fewer primitives