For such a small baby, I'd bring my own. They do have life vests, usually you can get them from the lifeguard. I've never seen life vests that are little enough for a baby. Only small children.
I got my life vests from the front desk at the Poly and the Contemporary. This is where they keep the kiddie sizes. My son was 8 months old at the time!