Handle hiding the @jit_kernel
rsuderman opened this issue · comments
Address comment linked below. jit_kernel is more an implementation detail and should not be exposed to the user. See about excluding it from import / reworking to be non-visible.
https://github.com/iree-org/iree-jax/blob/main/iree/jax/builtins.py#L46