#1 fixes, build jax