Whether in the ocean or a one-block deep pool, baby chickens die within a minute. Baby pigs and cows sink and die almost immediately in the ocean. Baby villagers survive to adulthood while bobbing in the ocean as aninmals should.
This report is similar to
MC-9388 but adds info about ocean, cows, pigs, and villagers.
MC-9388 Baby mobs cannot swim