iOS 17 is basically doing that: https://www.macrumors.com/guide/ios-17-keyboard/