Hi Alison, I doubt it's the food. Some Axolotls do just die - esp as this age.
But if it's on a large scale; i.e. one morning you've lost several - then it could be a bacterial infection. I've had this a couple of times - always from pet shop daphnia, so now I only use the stuff I catach/culture.
If it is a bacterial infection (only indicator will be large numbers dying) do a full water change.
If it's just one or two every few days, I wouldn't worry too much. Just keep an eye on them.