@retiolus
I've got this keybinding on my :neovim: config, it works for me.

vim.keymap.set("i", "jj", "<ESC>")

or

vim.keymap.set('i', 'jk', '<ESC>')