I have never used stock Xiaomi software for more than the time it took to unlock the bootloader. The custom ROMs for my devices were jank as fuck too. I will never buy from Xiaomi again. Only Pixel and GrapheneOS from now on.