Investigation under way after baby dies