I had my babies in individual containers. Initially (the stage you're at), I had only about 250 mL of water in each container. The advantage here is that you can see each animal as an individual so you know who's eating and who isn't. Also, a small container keeps the food close to the axies so there is a much better chance for them to get the food. As the axies grew I used a larger container (1 litre). Since the animals actually eat more of the shrimp you need to put fewer shrimp in each container to get the same effect.
Obviously, the smaller containers require much more diligence in regard to water quality.
Finally, there's no concern about cannibalism amongst your axies when they're kept separately.