There are more details about AI engine heating risk:
Thank you for your support of Xiaomi products. When you just get a phone/upgrade the version/change the phone, the phone will compile the newly updated code in a short period of time. This process will consume system resources and cause temporary power consumption and heating problems. This process will last for 4-5 days and is normal.
After just changing the phone, please update the software version. The software on the old phone is not compatible with the current Android underlying system. Please update the software or uninstall and reinstall it.
Does it help?