FUTO Keyboard

It's almost as good as Google's and extremely customizable; so you can bring your own AI 😎

I recommend installing through using F-Droid and testing out for yourself. The default AI is lightweight enough even for an old/budget smartphone. FUTO is better than any other open source solution I have ever come across for the job. Please consider financially supporting them.