When you create a world at the first and you go underwater, it is fine (no problem). When you exit the world and go back and go underwater, the screen become glitchy. You can move your character and when you go out in the water, the screen go back into normal.
Update your game to 1.16.0.57. This bug is fixed in that version.