I think that’s what Apple did with ios17. they just prefer to use words like machine learning instead of AI. It definitely is improved, I was able to finally switch back after using Gboard for years.