I’ve actually just discovered by accident that it is enabled and working but you need to highlight the text you want to proofread and then click the AI button, which is all very difficult on mobile (I kept clicking Cut by mistake) and not at all discoverable.
So my next question is how to get the toolbar button like they have on Meta?