const inputRef = useRef(null)

In onSubmitEditing prop, include this:

setTimeout(() => inputRef.current.focus(), 250);

Ex:

const onEnterKeyPressed = () => {
      ...
    setTimeout(() => inputRef.current.focus(), 100);    
       ...
}