Don't want to write this by hand? GodCraft sets minecraft:underwater_movement for you with a switch.Get GodCraftOverview
minecraft:underwater_movement is an entity component: it goes in the components object of a custom mob, or in one of its component groups. Sets how fast the entity moves when it is underwater.
Example
Inside the components object of BP/entities/<name>.json (values are the defaults or placeholders):
BP/entities/<name>.json
{
"minecraft:underwater_movement": {}
}