jax

6f8456e9 - Error out if compiler_options are specified in a nested jit. They can only be specified on the top level jit.