A baby usually refers to a very small child, especially one that has not yet learned to walk. In your context, I would say eg a small child. I'd just say eg Put your toys away.
New words, one handy idiom, and a 2-minute quiz — delivered to your inbox to keep your streak alive.