jax

54ac8795 - add clarification about linearize vs jvp+vmap