I had no issues with the stock keyboard. I had a few bugs because of it being the 1st boot, but those go away after a reboot.
Does the stock recovery actually wipe the cache? I click wipe cache in mi recovery and it doesn't seem to do anything.
Sent from my Xiaomi MI2S