jax

1ed60b3f - support stacked doubling (#3490)