Update `jit.rs` and use operators for JIT IR where approprate