jax

f94c2063 - back to original tree_mimomap optimizers solution