jax

98c7567a - add flag for logging when jit performs compilation