It's essential for me too, ended up just using SwiftKey / GBoard with the network permission revoked on GrapheneOS.

I switched from GBoard because it's predictions for Arabic were horrendous, made swipe typing unusable.