jax

66f784d3 - add a simpler vmap(tree_call) test