#1 better JAX.make