you can get little vests that have foam in them and as the get better at swimming you remove some of the foam peices Ive seen them in sports shops
or you can just get arm floaties thats what my DD uses but she is 2 and a half...DD2 just floats around in a ring thing with a seat