How to use vim (key-bindings) with Visual Studio Code vim extension

Add the following to settings.json (open the Command Pallete and search for “User Settings”):

"vim.insertModeKeyBindings": [
     {
         "before": ["j", "j"],
         "after": ["<esc>"]
     }
]

That should do it.