helion.inline_triton is nice, but it's not so convenient to actually take a @triton.jit function and use it from within a helion function.